ZeroPad2d¶

class dragon.vm.torch.nn.ZeroPad2d(padding)[source]¶

Pad input according to the last 2-dimensions with zeros.

The padded dimension is computed as:

\[\text{Dim}_{out} = \text{Dim}_{in} + pad_l + pad_r \]

Examples:

m = torch.nn.ZeroPad2d(1)
x = torch.randn(1, 2, 2)
y = m(x)  # (1, 2, 2) -> (1, 4, 4)

See also

torch.nn.functional.pad(…)