本文主要是介绍Pytorch_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')
in_channels
输入数据通道数out_channels
输出数据通道数kennel_size
卷积核大小,int或tuplestride
步长padding
每个维度零填充的数量dilation
内核点之间的距离,也称à trous algorithmgroups
控制inputs与outputs间的连接
groups=1
,所有输入都卷积到输出groups=2
,并排设置两个conv层,每个层查看一半的输入通道,并生成一半的输出通道,然后将两者连接起来groups=in_channels
,每个输入通道都有它自己的filter,size为[out_channels/in_channels]
这篇关于Pytorch_nn.Conv2d的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!