卷积神经网络 一些参数计算

卷积神经网络 一些参数计算,第1张


N 输入图片个数
C 输入图片的通道数
H 输入图片的高
W 输入图片的宽
N1 输入图片个数
C1 输入图片的通道数
H1 输入图片的高
W1 输入图片的宽
k 卷积核尺寸

1.进行一次卷积所计算的次数 KKC1CH1W1
2.进行一次卷积的参数 K
KC1C
3.通过卷积后输出结果的尺寸

例如:
1.一直某卷积层为nn.conv2d(in_channels=3, out_channels=6, kernel_size=5),输入张量为64X3x256x256(NCH*W)
1.1 试求出输出张量的size
1.2求出卷积层的权重参数数量
1.3求出整个卷积层的乘法计算次数

欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/langs/740177.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-28
下一篇 2022-04-28

发表评论

登录后才能评论

评论列表(0条)

保存