改变矩阵形状的代码

return F.linear(input, self.weight, self.bias)
RuntimeError: mat1 and mat2 shapes cannot be multiplied (249856x224 and 256x78)
报错解释:矩阵1 和 矩阵2无法相乘。矩阵1的维度是249856x224,矩阵2的维度是256×78.
在init()中,我调用nn.Linear,将维度从256,变为指定的78

 self.output = nn.Linear(in_features=256,out_features=78,bias=False)

所以,报错的256 × 78 指的就是self. output语句中的in_features和 output_feature

所以,我们应该使249856 × 224 变为 256 × blabla大小的矩阵。至于这个blabla的值到底是多少,我们留给系统自己算。
改变形状的代码如下所示:我在forward()里加上一句:

 out = out.view(-1,256)

out的形状改变后,继续调用self.output语句

out = out.view(-1,256)
out = self.output(out) # out:(200704,256)
out = self.relu(out)
return out

你可能感兴趣的:(矩阵,python,神经网络)