记录下常用参数使用,方便查阅
conv2d的官方文档点击这里
首先是原函数的参数:

torch.nn.Conv2d(in_channels, out_channels, kernel_size, 
stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros')

各个参数含义如下图:
Pytorch——conv2d参数使用
输入输出的格式:

input(N,C-in,H-in,W-in)
output(N,C-out,H-out,W-out)
  • N是batch的大小
  • C是通道数量
  • H是输入的高度
  • W是输入的宽度

其中N和C-in、C-out是人为指定,H,W是原始输入,H-out,W-out是通过公式计算出来的,公式如下:
Pytorch——conv2d参数使用

相关文章:

  • 2021-04-15
  • 2022-12-23
  • 2021-05-13
  • 2021-05-22
  • 2021-09-25
  • 2021-07-13
  • 2021-06-10
猜你喜欢
  • 2022-01-15
  • 2022-12-23
  • 2022-01-03
  • 2022-12-23
  • 2021-11-09
  • 2021-06-03
  • 2022-02-10
相关资源
相似解决方案