nn.ReflectionPad2d()
:利用输入边界的反射来填充输入张量。其填充方式为新的dim值使用反方向的最下边元素的值
这是在风格迁移中用到的操作。镜像填充的方式相比于前面使用固定数值进行填充,有可能获得更好的卷积结果
torch.nn.ReflectionPad2d(padding)
可以看出第一行第一列的4是原来右下角的4,第一行第二列的3原来左下角的3
Pytorch nn.ReflectionPad2d(镜像填充)
最新推荐文章于 2023-04-22 14:56:19 发布