ReflectionPad3d#
- class torch.nn.ReflectionPad3d(padding)[source]#
使用输入边界的反射来填充输入张量。
对于 N 维填充,请使用
torch.nn.functional.pad()
。- 参数
padding (int, tuple) – 填充大小。如果为 int,则在所有边界使用相同的填充。如果是 6-tuple,则使用 (, , , , , ) 注意,填充大小应小于相应的输入维度。
- 形状
输入: 或 。
输出: 或 ,其中
示例
>>> m = nn.ReflectionPad3d(1) >>> input = torch.arange(8, dtype=torch.float).reshape(1, 1, 2, 2, 2) >>> m(input) tensor([[[[[7., 6., 7., 6.], [5., 4., 5., 4.], [7., 6., 7., 6.], [5., 4., 5., 4.]], [[3., 2., 3., 2.], [1., 0., 1., 0.], [3., 2., 3., 2.], [1., 0., 1., 0.]], [[7., 6., 7., 6.], [5., 4., 5., 4.], [7., 6., 7., 6.], [5., 4., 5., 4.]], [[3., 2., 3., 2.], [1., 0., 1., 0.], [3., 2., 3., 2.], [1., 0., 1., 0.]]]]])