mindspore.ops.Padding

class mindspore.ops.Padding(pad_dim_size=8)[源代码]

将输入Tensor的最后一个维度从1扩展到 pad_dim_size ,其填充值为0。

参数:

  • pad_dim_size (int) - 指定填充的大小,待扩展的 x 的最后一个维度的值,必须为正数。默认值:8。

输入:

  • x (Tensor) - 输入Tensor,其shape为 \((x_1, x_2, ..., x_R)\)x 的秩必须至少为2。 x 的最后一个维度必须为1。数据类型为Number。

输出:

Tensor,其shape是 \((z_1, z_2, ..., z_N)\)

异常:

  • TypeError - pad_dim_size 不是int。

  • ValueError - pad_dim_size 小于1。

  • ValueError - x 的最后一个维度不等于1。

支持平台:

Ascend

样例:

>>> x = Tensor(np.array([[8], [10]]), mindspore.float32)
>>> pad_dim_size = 4
>>> output = ops.Padding(pad_dim_size)(x)
>>> print(output)
[[ 8.  0.  0.  0.]
 [10.  0.  0.  0.]]