mars.tensor.atleast_1d#

mars.tensor.atleast_1d(*tensors)[来源]#

将输入转换为至少具有一个维度的张量。

标量输入被转换为1维张量，而高维输入则被保留。

Parameters

  • tensors1 (array_like) – 一个或多个输入张量。

  • tensors2 (array_like) – 一个或多个输入张量。

  • ... (array_like) – 一个或多个输入张量。

Returns

ret – 一个张量，或张量的列表，每个都有 a.ndim >= 1。只有在必要时才会进行复制。

Return type

张量

另请参阅

atleast_2d, atleast_3d

示例

>>> import mars.tensor as mt

>>> mt.atleast_1d(1.0).execute()
array([ 1.])

>>> x = mt.arange(9.0).reshape(3,3)
>>> mt.atleast_1d(x).execute()
array([[ 0.,  1.,  2.],
       [ 3.,  4.,  5.],
       [ 6.,  7.,  8.]])
>>> mt.atleast_1d(x) is x
True

>>> mt.atleast_1d(1, [3, 4]).execute()
[array([1]), array([3, 4])]