1*1卷积层运行报错是怎么回事

动手学深度学习（Pytorch版）
1*1卷积层代码和书上一样，为什么会报错啊

• 这篇文章：Pytorch学习笔记（六）——1x1卷积核的理解 也许能够解决你的问题，你可以看下
• 除此之外, 这篇博客: 【动手学习pytorch笔记】9.卷积神经网络基础（卷积层，填充和步长，多输入输出通道，池化层）中的 1 * 1 卷积层 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  

• def corr2d_multi_in_out_1x1(X, K):
      c_i, h, w = X.shape
      c_o = K.shape[0]
      X = X.reshape((c_i, h * w))
      K = K.reshape((c_o, c_i))
      # 全连接层中的矩阵乘法
      Y = torch.matmul(K, X)
      return Y.reshape((c_o, h, w))
  
  X = torch.normal(0, 1, (3, 3, 3))
  K = torch.normal(0, 1, (2, 3, 1, 1))
  
  Y1 = corr2d_multi_in_out_1x1(X, K)
  Y2 = corr2d_multi_in_out(X, K)
  assert float(torch.abs(Y1 - Y2).sum()) < 1e-6
  

  相当于全连接层

• 您还可以看一下 龙良曲老师的深度学习与PyTorch入门实战教程课程中的 什么是梯度-2小节, 巩固相关知识点