mindspore.ops.scalar_to_tensor

mindspore.ops.scalar_to_tensor(input_x, dtype=mstype.float32)[源代码]

将Scalar转换为指定数据类型的 Tensor

参数:

  • input_x (Union[int, float]) - 输入是Scalar。只能是常量值。

  • dtype (mindspore.dtype) - 指定输出的数据类型。只能是常量值。默认值:mindspore.float32。

返回:

Tensor,0维Tensor,其值和输入一致。

异常:

  • TypeError - input_x 既不是int也不是float。

支持平台:

Ascend GPU CPU

样例:

>>> data = 1
>>> output = ops.scalar_to_tensor(data, mindspore.float32)
>>> print(output)
1.0