mindspore.ops.cast
- mindspore.ops.cast(input, dtype)[源代码]
转换输入tensor的数据类型。
说明
将复数转换为bool类型的时候,不考虑复数的虚部,只要实部不为零,返回
True
,否则返回False
。- 参数:
input (Union[Tensor, Number]) - 输入tensor或者数值型数据。
dtype (dtype.Number) - 转换后的数据类型。仅支持常量值。
- 返回:
Tensor
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> input = mindspore.tensor([1, 2, 3], mindspore.int32) >>> dtype = mindspore.float64 >>> output = mindspore.ops.cast(input, dtype) >>> print(output.dtype) Float64 >>> print(output) [1. 2. 3.]