ConstantPad3d¶
- class torch.nn.ConstantPad3d(padding, value)[源代码]¶
使用常数值填充输入张量的边界。
对于N维填充,使用
torch.nn.functional.pad()。- Shape:
输入: 或 .
输出: 或 , 其中
示例:
>>> m = nn.ConstantPad3d(3, 3.5) >>> input = torch.randn(16, 3, 10, 20, 30) >>> output = m(input) >>> # 对不同边使用不同的填充 >>> m = nn.ConstantPad3d((3, 3, 6, 6, 0, 1), 3.5) >>> output = m(input)