PyTorch
torch / nn / torch.nn
ReplicationPad3d¶
-
class
torch.nn.
ReplicationPad3d
(padding: Union[T, Tuple[T, T, T, T, T, T]])[source]¶ Pads the input tensor using replication of the input boundary.
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.ReplicationPad3d(3) >>> input = torch.randn(16, 3, 8, 320, 480) >>> output = m(input) >>> # using different paddings for different sides >>> m = nn.ReplicationPad3d((3, 3, 6, 6, 1, 1)) >>> output = m(input)
此页内容是否对您有帮助
感谢反馈!