ReplicationPad2d#
- classtorch.nn.modules.padding.ReplicationPad2d(padding)[source]#
Pads the input tensor using replication of the input boundary.
ForN-dimensional padding, use
torch.nn.functional.pad().- Parameters
padding (int,tuple) – the size of the padding. If isint, uses the samepadding in all boundaries. If a 4-tuple, uses (,,,)Note that the output dimensions must remain positive.
- Shape:
Input: or.
Output: or, where
Examples:
>>>m=nn.ReplicationPad2d(2)>>>input=torch.arange(9,dtype=torch.float).reshape(1,1,3,3)>>>inputtensor([[[[0., 1., 2.], [3., 4., 5.], [6., 7., 8.]]]])>>>m(input)tensor([[[[0., 0., 0., 1., 2., 2., 2.], [0., 0., 0., 1., 2., 2., 2.], [0., 0., 0., 1., 2., 2., 2.], [3., 3., 3., 4., 5., 5., 5.], [6., 6., 6., 7., 8., 8., 8.], [6., 6., 6., 7., 8., 8., 8.], [6., 6., 6., 7., 8., 8., 8.]]]])>>># using different paddings for different sides>>>m=nn.ReplicationPad2d((1,1,2,0))>>>m(input)tensor([[[[0., 0., 1., 2., 2.], [0., 0., 1., 2., 2.], [0., 0., 1., 2., 2.], [3., 3., 4., 5., 5.], [6., 6., 7., 8., 8.]]]])