torch.nn.conv2d详细参数
时间: 2023-09-23 13:06:09 浏览: 96
pytorch1.0中torch.nn.Conv2d用法详解
torch.nn.Conv2d类的常用参数如下:
- in_channels: 输入张量中的通道数。
- out_channels: 输出张量中的通道数,即卷积核的数量。
- kernel_size: 卷积核的尺寸,可以是一个整数或一个元组,如(3,3)。
- stride: 卷积核的步幅,可以是一个整数或一个元组,如(1,2)。
- padding: 输入张量四周的填充数,可以是一个整数或一个元组,如(1,1)。
- dilation: 卷积核元素之间的间距,可以是一个整数或一个元组,如(2,2)。
- groups: 输入和输出通道之间的连接数,可以是一个整数或默认值1。
- bias: 是否添加偏置项,默认为True。
以上参数中,in_channels、out_channels、kernel_size和stride是必须指定的参数,其他参数都有默认值。
阅读全文