【error】RuntimeError: size mismatch與全連接fc層

今天跑代碼的時候遇到了這個錯誤： RuntimeError: size mismatch, m1:[1152 x 1] ,m2:[576 x 192] ,at /opt/conda/conda-bld/pytorch_1524 調試之後發現是如下有問題： 源代碼在這裏： 這是class裏init中相應的部分 def forward的相應的問題 問題就在於y1 = self.fc_1(y)這裏，fc

>>阅读原文<<