mindspore.ops.Padding
- class mindspore.ops.Padding(pad_dim_size=8)[源代码]
将输入Tensor的最后一个维度从1扩展到 pad_dim_size ,其填充值为0。
更多参考详见
mindspore.ops.padding()
。- 参数:
pad_dim_size (int,可选) - 指定填充的大小,待扩展的 x 的最后一个维度的值,必须为正数。默认值:
8
。
- 输入:
x (Tensor) - 输入Tensor,二维或者更高维Tensor, x 的最后一个维度必须为1。数据类型为Number。
- 输出:
填充后的Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> 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.]]