mindspore.numpy.atleast_3d

mindspore.numpy.atleast_3d(*arys)[源代码]

将输入重新调整为至少三维的数组。

说明

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

参数：

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

返回：

Tensor或Tensor列表，每个Tensor满足 \(a.ndim >= 3\) 。例如，一个shape为 (N,) 的一维数组会变为shape为 (1, N, 1) 的Tensor，一个shape为 (M, N) 的二维数组会变为shape为 (M, N, 1) 的Tensor。

异常：

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_3d(a, b, c)
>>> print(output)
    [Tensor(shape=[2, 3, 1], 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, 1], dtype=Float32, value= [[[1.00000000e+00]]]),
    Tensor(shape=[1, 5, 1], dtype=Float32,
    value= [[[1.00000000e+00], [1.00000000e+00], [1.00000000e+00],
    [1.00000000e+00], [1.00000000e+00]]])]