mindspore.ops.TupleToArray

class mindspore.ops.TupleToArray[源代码]

将tuple转换为Tensor。

如果tuple中第一个数据类型为int,则输出Tensor的数据类型为int。否则,输出Tensor的数据类型为float。

输入:

  • input_x (tuple) - 数值型组成的tuple。其元素具有相同的类型。仅支持常量值。shape为 \((N,*)\) ,其中 \(*\) 表示任意数量的附加维度。

输出:

Tensor。如果输入tuple包含 N 个数值型元素,则输出Tensor的shape为(N,)。

异常:

  • TypeError - input_x 不是tuple。

  • ValueError - input_x 的长度小于或等于0。

支持平台:

Ascend GPU CPU

样例:

>>> input_x = (1,2,3)
>>> print(type(input_x))
<class 'tuple'>
>>> output = ops.TupleToArray()(input_x)
>>> print(type(output))
<class 'mindspore.common.tensor.Tensor'>
>>> print(output)
[1 2 3]