RuntimeError: size mismatch, m1: [80 x 4], m2: [320 x 50] at …\aten\src\TH/generic/THTensorMath.cpp:41

使用pytorch進行深度學習的訓練會出現這種問題，原因是fc全連接層的輸入維度問題，由于輸入是二維的數據，很多時候在輸入全連接層的時候我們沒有調整維度
例如：

def forward(self, input):x = self.pool1(F.relu(self.conv1(input)))x = self.pool2(F.relu(self.conv2(x)))x = self.fc1(x)x = self.fc2(x)return x

我們的fc1的輸入將是804的情況，而需要的是3201的情況，這時會報錯

改為：

def forward(self, input):x = self.pool1(F.relu(self.conv1(input)))x = self.pool2(F.relu(self.conv2(x))).view(320)x = self.fc1(x)x = self.fc2(x)return x

使用view調整維度即可

總結

以上是生活随笔為你收集整理的RuntimeError: size mismatch, m1: [80 x 4], m2: [320 x 50] at ..\aten\src\TH/generic/THTensorMath.cpp的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。