mindspore.ops.atleast_2d

mindspore.ops.atleast_2d(inputs)

当输入中的tensor小于二维时，返回它的二维tensor。其他维度的tensor，直接返回原tensor。

参数：

• inputs (Union[Tensor, list[Tensor]]) - 输入tensor或tensor列表。

返回：

tensor或tensor列表。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> # case 1: Input is a zero-dimensional tensor.
>>> x = mindspore.tensor(1)
>>> mindspore.ops.atleast_2d(x)
Tensor(shape=[1, 1], dtype=Int64, value= [[1]])
>>>
>>> # case 2: Input is a 2-dimensional tensor.
>>> y = mindspore.tensor([[0, 1], [2, 3]])
>>> mindspore.ops.atleast_2d(y)
Tensor(shape=[2, 2], dtype=Int64, value=
[[0, 1],
 [2, 3]])
>>>
>>> # case 3: Input is a list containing tensors of various dimensions.
>>> mindspore.ops.atleast_2d([x, y])
(Tensor(shape=[1, 1], dtype=Int64, value=
 [[1]]),
 Tensor(shape=[2, 2], dtype=Int64, value=
 [[0, 1],
  [2, 3]]))