菜鸟学院
PyTorch笔记 卷积层 torch.nn.Conv2d
时间 2021-06-09
标签
PyTorch
pytorch
官方文档:
CONV2D
kernel_size:卷积核边长
stride:卷积核移动步长
padding:在输入周围补一圈padding
一、两种调用方式
第1种:
第2种:
注意,建议使用第2种
二、属性