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)