mindspore.mint.nn.ZeroPad3d

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

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

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 shape.

Supported Platforms:: Ascend

Examples

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