PyTorch中的nn.Conv2d
一般来说,二维卷积nn.Conv2d用于图像数据,对宽度和高度都进行卷积
Conv2d 参数详解
torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode=‘zeros’, device=None, dtype=None)
Parameters
in_channels (int) – Number of chann
原创
2021-09-21 21:57:32 ·
437 阅读 ·
0 评论