mindspore.ops.atleast_1d
- mindspore.ops.atleast_1d(inputs)[源代码]
调整 inputs 中的Tensor维度,使输入中每个Tensor维度不低于1。
Scalar输入被转换为一维Tensor,而高维输入则被保留。
- 参数:
inputs (Union[Tensor, list[Tensor]]) - 一个或多个输入Tensor。
- 返回:
Tensor或Tensor列表。如果返回列表,则列表中的每一个元素 a 满足: a.ndim >= 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_1d([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.]