mindspore.ops.unsqueeze
- mindspore.ops.unsqueeze(input, dim)[源代码]
对输入 input 在给定维上添加额外维度。
扩展后的Tensor中位置对应 dim 的维度为插入的新维度。
- 参数:
input (Tensor) - 输入Tensor,shape为 \((n_1, n_2, ..., n_R)\)。
dim (int) - 新插入的维度的位置。 dim 的值必须在范围 [-input.ndim-1, input.ndim] 内。仅接受常量输入。
- 返回:
Tensor,维度在指定轴扩展之后的Tensor,与 input 的数据类型相同。如果 dim 是0,那么它的shape为 \((1, n_1, n_2, ..., n_R)\)。
- 异常:
TypeError - 如果 dim 不是int。
ValueError - 如果 dim 超出了 \([-input.ndim-1, input.ndim]\) 的范围。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> input_tensor = Tensor(np.array([[2, 2], [2, 2]]), mindspore.float32) >>> output = ops.unsqueeze(input_tensor, dim=0) >>> print(output) [[[2. 2.] [2. 2.]]]