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