torch 卷积神经网络构建(3)

文章目录

  • 笔记
  • 特征图计算公式

笔记

torch 卷积神经网络构建(3)_第1张图片

特征图计算公式

(h - ksize + 2* padding)/stride + 1

(28 - 5 + 4)/1 + 1 = 28 * 28 * 16

torch 卷积神经网络构建(3)_第2张图片
torch 卷积神经网络构建(3)_第3张图片
池化层(MaxPool)输出大小计算
O = (I - P) / S +1

torch 卷积神经网络构建(3)_第4张图片
数据读取
torch 卷积神经网络构建(3)_第5张图片

你可能感兴趣的:(torch,cnn,深度学习,人工智能)