mindspore.ops.max_unpool3d
- mindspore.ops.max_unpool3d(x, indices, kernel_size, stride=None, padding=0, output_size=None)[source]
Computes the inverse of
mindspore.ops.max_pool3d()
.max_unpool3d keeps the maximal value and set all position of non-maximal values to zero. Typically the input is of shape
or , and the output is of shape or . The operation is as follows.- Parameters
x (Tensor) – The input Tensor to invert. Tensor of shape
or .indices (Tensor) – Max values' index represented by the indices. Tensor of shape must be same with input 'x'. Values of indices must belong to
. Data type must be in int32 or int64.kernel_size (Union[int, tuple[int]]) – The size of kernel used to take the maximum value, an int number that represents depth, height and width of the kernel, or a tuple of three int numbers that represent depth, height and width respectively.
stride (Union[int, tuple[int]]) – The distance of kernel moving, an int number that represents the depth, height and width of movement are both stride, or a tuple of three int numbers that represent depth, height and width of movement respectively. Default:
None
, which indicates the moving step is kernel_size .padding (Union[int, tuple[int]]) – The pad value to be filled. Default:
0
. If padding is an integer, the paddings of depth, height and width are the same, equal to padding. If padding is a tuple of three integers, the padding of depth, height and width equal to padding[0], padding[1] and padding[2] correspondingly.output_size (tuple[int], optional) – The output size. Default:
None
. If output_size == (), then the shape of output computed by kernel_size, stride and padding. If output_size != (), then output_size must be or or and output_size must belong to .
- Returns
Tensor, with shape
or , with the same data type with x.- Raises
TypeError – If data type of x or indices is not supported.
TypeError – If kernel_size, stride or padding is neither an int nor a tuple.
ValueError – If numbers in stride or padding (also support 0 and (0, 0, 0)) or kernel_size is not positive.
ValueError – If the shape of x and indices are not equal.
ValueError – If kernel_size, stride or padding is a tuple whose length is not equal to 3.
ValueError – If x whose length is not 4 or 5.
ValueError – If output_size whose length is not 0, 4 or 5.
ValueError – If output_size whose type is not tuple.
ValueError – If output_size is not close to output size computed by attr kernel_size, stride, padding.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import numpy as np >>> from mindspore import Tensor, ops >>> x = Tensor(np.array([[[[[0, 1], [8, 9]]]]]).astype(np.float32)) >>> indices= Tensor(np.array([[[[[0, 1], [2, 3]]]]]).astype(np.int64)) >>> output = ops.max_unpool3d(x, indices, kernel_size=2, stride=1, padding=0) >>> print(output) [[[[[0. 1. 8.] [9. 0. 0.] [0. 0. 0.]] [[0. 0. 0.] [0. 0. 0.] [0. 0. 0.]]]]]