torch.nn.conv2d 参数padding_mode
时间: 2023-09-13 16:05:33 浏览: 549
基于torch.nn的回归、二分类、多分类任务代码
`padding_mode`是`torch.nn.conv2d`中的一个参数,用于设置在卷积操作中如何填充图像的边缘区域。
有两个选项可以选择:
- `'zeros'`:使用零来填充图像的边缘区域。
- `'reflect'`:使用图像的对称副本来填充边缘区域,这样可以避免出现人工填充时可能出现的边缘效应。
默认值为`'zeros'`。
阅读全文