PyTorch
torch / nn / torch.nn
ConstantPad3d¶
-
class
torch.nn.
ConstantPad3d
(padding: Union[T, Tuple[T, T, T, T, T, T]], value: float)[source]¶ Pads the input tensor boundaries with a constant value.
For N-dimensional padding, use
torch.nn.functional.pad()
.- Parameters
padding (int, tuple) – the size of the padding. If is int, uses the same padding in all boundaries. If a 6-tuple, uses (padding_left , padding_right , padding_top , padding_bottom , padding_front , padding_back )
- Shape:
Input: (N,C,Din,Hin,Win)
Output: (N,C,Dout,Hout,Wout) where
Dout=Din+padding_front+padding_back
Hout=Hin+padding_top+padding_bottom
Wout=Win+padding_left+padding_right
Examples:
>>> m = nn.ConstantPad3d(3, 3.5) >>> input = torch.randn(16, 3, 10, 20, 30) >>> output = m(input) >>> # using different paddings for different sides >>> m = nn.ConstantPad3d((3, 3, 6, 6, 0, 1), 3.5) >>> output = m(input)
此页内容是否对您有帮助
感谢反馈!