nn.con2d()参数
时间: 2024-06-12 19:10:17 浏览: 66
nn.Conv2d()是PyTorch中用于实现二维卷积操作的函数,其常用参数如下:
- in_channels:输入图像的通道数
- out_channels:输出图像的通道数
- kernel_size:卷积核的大小,可以是一个整数或者一个元组,如(3, 3)
- stride:卷积核的步长,可以是一个整数或者一个元组,如(1, 1)
- padding:输入图像四周补0的层数,可以是一个整数或者一个元组,如(1, 1)
- dilation:卷积核元素之间的间距,可以是一个整数或者一个元组,如(1, 1)
- groups:输入和输出之间连接的数量,可以是一个整数,默认值为1
- bias:是否添加偏置项,可以是一个布尔值,默认值为True
阅读全文