ReflectionPad1d¶
- class torch.nn.ReflectionPad1d(padding)[源代码]¶
使用输入边界的反射对输入张量进行填充。
对于 N 维填充,请使用
torch.nn.functional.pad()
。- 形状
输入: 或 .
输出: 或 ,其中
示例
>>> m = nn.ReflectionPad1d(2) >>> input = torch.arange(8, dtype=torch.float).reshape(1, 2, 4) >>> input tensor([[[0., 1., 2., 3.], [4., 5., 6., 7.]]]) >>> m(input) tensor([[[2., 1., 0., 1., 2., 3., 2., 1.], [6., 5., 4., 5., 6., 7., 6., 5.]]]) >>> # using different paddings for different sides >>> m = nn.ReflectionPad1d((3, 1)) >>> m(input) tensor([[[3., 2., 1., 0., 1., 2., 3., 2.], [7., 6., 5., 4., 5., 6., 7., 6.]]])