mindspore.mint.unsqueeze

mindspore.mint.unsqueeze(input, dim)[源代码]

在输入tensor的指定维度上增加一维。

参数：

input (Tensor) - 输入tensor。
dim (int) - 指定维度。

返回：

Tensor

支持平台：

Ascend

样例：

>>> import mindspore
>>> input = mindspore.tensor([1, 2, 3, 4])
>>> mindspore.mint.unsqueeze(input, 0)
Tensor(shape=[1, 4], dtype=Int64, value=
[[1, 2, 3, 4]])
>>> mindspore.mint.unsqueeze(input, 1)
Tensor(shape=[4, 1], dtype=Int64, value=
[[1],
 [2],
 [3],
 [4]])