mindspore.ops.atleast_3d
- mindspore.ops.atleast_3d(inputs)[源代码]
调整 inputs 中的Tensor维度,使输入中每个Tensor维度不低于3。
Scalar、一维或二维Tensor被转换为三维Tensor,而高维输入则被保留。
- 参数:
inputs (Union[Tensor, list[Tensor]]) - 一个或多个输入Tensor。
- 返回:
Tensor或Tensor列表。如果返回列表,则列表中的每一个元素 a 满足: a.ndim >= 3。 例如,一个shape为 \((N,)\) 的Tensor,操作后shape变成 \((1, N, 1)\) ,shape为 \((M, N)\) 的2-D Tensor shape变成 \((M, N, 1)\) 。
- 异常:
TypeError - inputs 不是一个Tensor或Tensor列表。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import numpy as np >>> from mindspore import Tensor, ops >>> x1 = Tensor(np.ones((2, 3))) >>> x2 = Tensor(np.ones(())) >>> x3 = Tensor(np.ones(5)) >>> out = ops.atleast_3d([x1, x2, x3]) >>> print(out[0].asnumpy()) [[[1.] [1.] [1.]] [[1.] [1.] [1.]]] >>> print(out[1].asnumpy()) [[[1.]]] >>> print(out[2].asnumpy()) [[[1.] [1.] [1.] [1.] [1.]]]