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