mindspore.mint.nn.ZeroPad2d

class mindspore.mint.nn.ZeroPad2d(padding)[source]

Pad the last 2 dimension of input tensor with 0 using padding. The padding mode is: (padding_left, padding_right, padding_top, padding_bottom).

For more information, please refer to mindspore.mint.nn.functional.pad().

Warning

This is an experimental API that is subject to change or deletion.

Parameters: padding (Union[int, tuple, list]) – Specifies padding size.

Inputs:

input (Tensor) - shape is \((N, *)\), where \(*\) means, any number of additional dimensions.

Outputs:

Tensor, the tensor after padding.

Raises

ValueError – If padding contains negative value.
ValueError – If padding is a tuple or list, and the length does not match the tensor dimension.

Supported Platforms:: Ascend

Examples

>>> import mindspore
>>> x = mindspore.mint.ones((1, 2, 3, 4))
>>> padding = 1
>>> pad2d = mindspore.mint.nn.ZeroPad2d(padding)
>>> out = pad2d(x)
>>> out.shape
(1, 2, 5, 6)
>>> padding = (1, 1, 0, 1)
>>> pad2d = mindspore.mint.nn.ZeroPad2d(padding)
>>> out = pad2d(x)
>>> out.shape
(1, 2, 4, 6)