RuntimeError:mat1 和 mat2 形状不能相乘(25×340 和 360×1)

社会演员多 pytorch 2655

原文标题RuntimeError: mat1 and mat2 shapes cannot be multiplied (25×340 and 360×1)

我收到此错误消息,但不知道为什么。我的输入是来自表格数据的 (batch, 1, 312),这个 CNN 是为回归预测而构建的。我使用公式 (输入 + 2*padding – filter size)/stride + 1 如下评论所示。问题出现在 x = self.fc(x) 处,我不知道为什么。非常感谢您的帮助。谢谢。

class CNNWeather(nn.Module):
    # input (batch, 1, 312)
    def __init__(self):
        super(CNNWeather, self).__init__()
        self.conv1 = nn.Conv1d(in_channels=1, out_channels=8, kernel_size=9, stride=1, padding='valid')         # (312+2*0-9)/1 + 1 = 304
        self.pool1 = nn.AvgPool1d(kernel_size=2, stride=2)                                                      # 304/2 = 302
        self.conv2 = nn.Conv1d(in_channels=8, out_channels=12, kernel_size=3, stride=1, padding='valid')        # (302-3)/1+1 = 300
        self.pool2 = nn.AvgPool1d(kernel_size=2, stride=2)                                                      # 300/2 = 150
        self.conv3 = nn.Conv1d(in_channels=12, out_channels=16, kernel_size=3, stride=1, padding='valid')       # (150-3)/1+1 = 76
        self.pool3 = nn.AvgPool1d(kernel_size=2, stride=2)                                                      # 76/2 = 38
        self.conv4 = nn.Conv1d(in_channels=16, out_channels=20, kernel_size=3, stride=1, padding='valid')       # (38-3)/1+1 = 36
        self.pool4 = nn.AvgPool1d(kernel_size=2, stride=2)                                                      # 36/2 = 18  (batch, 20, 18)
        self.fc = nn.Linear(in_features=20*18, out_features=1)

    def forward(self, x):
        x = self.pool1(F.relu(self.conv1(x)))
        x = self.pool2(F.relu(self.conv2(x)))
        x = self.pool3(F.relu(self.conv3(x)))
        x = self.pool4(F.relu(self.conv4(x)))
        print(x.size())
        x = x.view(x.size(0), -1)               # flatten   (batch, 20*18)
        x = self.fc(x)
        return x

原文链接:https://stackoverflow.com//questions/71423498/runtimeerror-mat1-and-mat2-shapes-cannot-be-multiplied-25×340-and-360×1

回复

我来回复
  • aaossa的头像
    aaossa 评论

    问题似乎与您的 FC 层的输入大小有关:

    self.fc = nn.Linear(in_features=20*18, out_features=1)
    

    上一层的输出是 340,所以必须使用 in_features=340 。

    这些是第三层和第四层的输出形状。

    torch.Size([5, 16, 73]) conv3 out
    torch.Size([5, 16, 36]) pool3 out
    torch.Size([5, 20, 34]) conv4 out
    torch.Size([5, 20, 17]) pool4 out
    

    请注意,“pool4”层中有 20×17,即 340 个元素。

    1年前 0条评论