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