mindspore.numpy.atleast_1d

mindspore.numpy.atleast_1d(*arys)

将输入转换为至少一维的数组。 标量输入将被转换为一维数组，而高维输入则保持不变。

说明

在图模式下，返回的不是Tensor列表，而是Tensor的tuple。

参数：

• *arys (Tensor) - 一个或多个输入Tensor。

返回：

Tensor或Tensor列表，每个Tensor满足 \(a.ndim >= 1\) 。

异常：

• TypeError - 如果输入不是Tensor。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> a = np.ones((2, 3))
>>> b = np.ones(())
>>> c = np.ones(5)
>>> output = np.atleast_1d(a, b, c)
>>> print(output)
    [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], dtype=Float32, value= [1.00000000e+00]),
    Tensor(shape=[5], dtype=Float32,
    value= [1.00000000e+00, 1.00000000e+00, 1.00000000e+00,
    1.00000000e+00, 1.00000000e+00])]