mindspore.ops.Cast

class mindspore.ops.Cast[源代码]

转换输入Tensor的数据类型。

输入:

  • input_x (Union[Tensor, Number]) - 输入要进行数据类型转换的Tensor,其shape为 \((x_1, x_2, ..., x_R)\)

  • type (dtype.Number) - 指定转换的数据类型。仅支持常量值。

输出:

Tensor,其shape与 input_x 相同,即 \((x_1, x_2, ..., x_R)\)

异常:

  • TypeError - input_x 既不是Tensor也不是数值型。

  • TypeError - type 不是数值型。

支持平台:

Ascend GPU CPU

样例:

>>> input_np = np.random.randn(2, 3, 4, 5).astype(np.float32)
>>> input_x = Tensor(input_np)
>>> type_dst = mindspore.int32
>>> cast = ops.Cast()
>>> output = cast(input_x, type_dst)
>>> print(output.dtype)
Int32
>>> print(output.shape)
(2, 3, 4, 5)