本文主要是介绍Pytorch——Conv2d、conv2d,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
Conv2d、conv2d是pytorch中进行卷积操作的2个类,虽然只是首字母大小写不同,使用起来方法也不一样,一个是类,一个是函数。
1 Conv2d
Conv2d是torch.nn中的类
1.1 初始化
CLASS
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)
参数说明
-
in_channels (int) – 输入图像通道
-
out_channels (int) – 输出图像通道,也就是用了多少层卷积核
-
kernel_size (int or tuple) – 卷积核大小
-
stride (int or tuple, optional) – 步长. Default
这篇关于Pytorch——Conv2d、conv2d的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!