mindspore.ops.atleast_2d

查看源文件
mindspore.ops.atleast_2d(inputs)[源代码]

调整 inputs 中的Tensor维度,使输入中每个Tensor维度不低于2。

Scalar或一维Tensor被转换为二维Tensor,而高维输入则被保留。

参数:
  • inputs (Union[Tensor, list[Tensor]]) - 一个或多个输入Tensor。

返回:

Tensor或Tensor列表。如果返回列表,则列表中的每一个元素 a 满足: a.ndim >= 2。

异常:
  • TypeError - inputs 不是一个Tensor或Tensor列表。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore.numpy as np
>>> from mindspore import ops
>>> x1 = np.ones((2, 3))
>>> x2 = np.ones(())
>>> x3 = np.ones(5)
>>> out = ops.atleast_2d([x1, x2, x3])
>>> print(out)
(Tensor(shape=[2, 3], dtype=Float32, value=
[[ 1.00000000e+00, 1.00000000e+00, 1.00000000e+00],
[ 1.00000000e+00, 1.00000000e+00, 1.00000000e+00]]), Tensor(shape=[1, 1], dtype=Float32, value=
[[ 1.00000000e+00]]), Tensor(shape=[1, 5], dtype=Float32, value=
[[ 1.00000000e+00, 1.00000000e+00, 1.00000000e+00, 1.00000000e+00, 1.00000000e+00]]))