mindspore.ops.max_unpool1d
- mindspore.ops.max_unpool1d(x, indices, kernel_size, stride=None, padding=0, output_size=None)[源代码]
maxpool1d 的部分逆过程。 maxpool1d 不是完全可逆的,因为非最大值丢失。 max_unpool1d 以 maxpool1d 的输出为输入,包括最大值的索引。在计算 maxpool1d 部分逆的过程中,非最大值设置为零。 支持的输入数据格式为 \((N, C, H_{in})\) 或 \((C, H_{in})\) ,输出数据的格式为 \((N, C, H_{out})\) 或 \((C, H_{out})\) ,计算公式如下:
\[\begin{split}\begin{array}{ll} \\ H_{out} = (H{in} - 1) \times stride[0] - 2 \times padding[0] + kernel\_size[0] \\ \end{array}\end{split}\]- 参数:
x (Tensor) - 待求逆的Tensor。shape为 \((N, C, H_{in})\) 或 \((C, H_{in})\) 。
indices (Tensor) - 最大值的索引。shape必须与输入 x 相同。取值范围需满足 \([0, H_{in} - 1]\) 。 数据类型必须是int32或int64。
kernel_size (Union[int, tuple[int]]) - 池化核尺寸大小。
stride (Union[int, tuple[int]]) - 池化操作的移动步长,若取值为 ‘0’ , ‘(0)’ 或者None, stride 值与 kernel_size 相同。默认值:None。
padding (Union[int, tuple[int]]) - 填充值。默认值:0。
output_size (tuple[int], 可选) - 输出shape。默认值:None。 如果output_size为(),那么输出shape根据 kernel_size 、 stride 和 padding 计算得出。 如果output_size不为(),那么 output_size 必须满足格式 \((N, C, H)\) 或 \((C, H)\) ,取值范围需满足: \([(N, C, H_{out} - stride[0]), (N, C, H_{out} + stride[0])]\) 。
- 返回:
shape为 \((N, C, H_{out})\) 或 \((C, H_{out})\) 的Tensor,数据类型与输入 x 相同。
- 异常:
TypeError - x 或 indices 的数据类型不支持。
TypeError - kernel_size , stride 或 padding 既不是整数也不是tuple。
ValueError - stride , padding 或 kernel_size 的值不是非负的。
ValueError - x 和 indices 的shape不一致。
ValueError - x 的长度不是2或3。
ValueError - output_size 的类型不是tuple。
ValueError - output_size 的长度不为0、2或3。
ValueError - output_size 的取值与根据 kernel_size , stride , padding 计算得到的结果差距太大。
- 支持平台:
GPU
CPU
样例:
>>> x = Tensor(np.array([[2, 4, 6, 8]]).astype(np.float32)) >>> indices = Tensor(np.array([[1, 3, 5, 7]]).astype(np.int64)) >>> output = ops.max_unpool1d(x, indices, kernel_size =2, stride=2, padding=0) >>> print(output.asnumpy()) [[0, 2, 0, 4, 0, 6, 0, 8]]