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
>>> import numpy as np
>>> input_np = np.random.randn(2, 3, 4, 5).astype(np.float32)
>>> input = mindspore.tensor(input_np)
>>> dtype = mindspore.int32
>>> output = mindspore.ops.cast(input, dtype)
>>> print(output.dtype)
Int32
>>> print(output.shape)
(2, 3, 4, 5)