mindspore.ops.padding
- mindspore.ops.padding(x, pad_dim_size=8)[源代码]
通过填充0,将输入张量的最后一个维度从1扩展到指定大小。
参数:
x (Tensor) - x 的shape为 \((x_1, x_2, ..., x_R)\),秩至少为2,它的最后一个维度必须为1。其数据类型为数值型。
pad_dim_size (int) - 要扩展的 x 的最后一个维度的值,该值必须为正数。默认值:8。
返回:
Tensor,其数据类型和维度必须和输入中的 x 保持一致。
异常:
TypeError - pad_dim_size 的数据类型不是int。
ValueError - pad_dim_size 的值小于1。
ValueError - x 的最后一个维度不等于1。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> from mindspore.ops import functional as F >>> x = Tensor(np.array([[8], [10]]), mindspore.float32) >>> pad_dim_size = 4 >>> output = F.padding(x, pad_dim_size) >>> print(output) [[ 8. 0. 0. 0.] [10. 0. 0. 0.]]