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]]))