mindspore.ops.ScalarToTensor
- class mindspore.ops.ScalarToTensor[源代码]
将Scalar转换为指定数据类型的 Tensor 。
输入:
input_x (Union[int, float]) - 输入是Scalar。只能是常量值。
dtype (mindspore.dtype) - 指定输出的数据类型。只能是常量值。默认值:mindspore.float32。
输出:
Tensor,0维Tensor,其值和输入一致。
异常:
TypeError - input_x 既不是int也不是float。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> op = ops.ScalarToTensor() >>> data = 1 >>> output = op(data, mindspore.float32) >>> print(output) 1.0