CNN的讲解可以看我的这篇文章CS231n 笔记:通俗理解 CNN
下面主要讲解一下如何用pytorch实现卷积神经网络
# 卷积神经网络的编写要用到nn.Conv2d
# 该API意为进行2D的函数卷积层计算...= nn.Conv2d(1, 5, kernel_size=3, stride=1, padding=0)
# 1代表每个kernel的chanel是1,5代表kernel的数量,同时也是输出到下一层的...[1, 5, 26, 26])
torch.Size([5, 1, 3, 3])
torch.Size([5])
这里可能需要解释一下weight的size,首先第一值5是kernel的数量,也是下一层的...(1, 3, 28, 28) # 1张图片,3chanel,28*28
# 由于输入图片的chanel是3,而kernel的chanel是2,两者不等,所以会报错
除此之外,pytorch还有一种定义卷积神经网络的方法