ZeroPad2d#

classtorch.nn.ZeroPad2d(padding)[source]#

Pads the input tensor boundaries with zero.

ForN-dimensional padding, usetorch.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 ($\text{padding\_left}$padding_left,$\text{padding\_right}$padding_right,$\text{padding\_top}$padding_top,$\text{padding\_bottom}$padding_bottom)

Shape:

• Input:$(N,C,H_{in},W_{in})$(N,C,Hin​,Win​) or$(C,H_{in},W_{in})$(C,Hin​,Win​).

• Output:$(N,C,H_{out},W_{out})$(N,C,Hout​,Wout​) or$(C,H_{out},W_{out})$(C,Hout​,Wout​), where

  $H_{out}=H_{in}+\text{padding\_top}+\text{padding\_bottom}$Hout​=Hin​+padding_top+padding_bottom

  $W_{out}=W_{in}+\text{padding\_left}+\text{padding\_right}$Wout​=Win​+padding_left+padding_right

Examples:

>>>m=nn.ZeroPad2d(2)>>>input=torch.randn(1,1,3,3)>>>inputtensor([[[[-0.1678, -0.4418,  1.9466],          [ 0.9604, -0.4219, -0.5241],          [-0.9162, -0.5436, -0.6446]]]])>>>m(input)tensor([[[[ 0.0000,  0.0000,  0.0000,  0.0000,  0.0000,  0.0000,  0.0000],          [ 0.0000,  0.0000,  0.0000,  0.0000,  0.0000,  0.0000,  0.0000],          [ 0.0000,  0.0000, -0.1678, -0.4418,  1.9466,  0.0000,  0.0000],          [ 0.0000,  0.0000,  0.9604, -0.4219, -0.5241,  0.0000,  0.0000],          [ 0.0000,  0.0000, -0.9162, -0.5436, -0.6446,  0.0000,  0.0000],          [ 0.0000,  0.0000,  0.0000,  0.0000,  0.0000,  0.0000,  0.0000],          [ 0.0000,  0.0000,  0.0000,  0.0000,  0.0000,  0.0000,  0.0000]]]])>>># using different paddings for different sides>>>m=nn.ZeroPad2d((1,1,2,0))>>>m(input)tensor([[[[ 0.0000,  0.0000,  0.0000,  0.0000,  0.0000],          [ 0.0000,  0.0000,  0.0000,  0.0000,  0.0000],          [ 0.0000, -0.1678, -0.4418,  1.9466,  0.0000],          [ 0.0000,  0.9604, -0.4219, -0.5241,  0.0000],          [ 0.0000, -0.9162, -0.5436, -0.6446,  0.0000]]]])

extra_repr()[source]#

Return the extra representation of the module.

Return type

str