cast

paddle. cast ( x, dtype ) [源代码]

x 的数据类型转换为 dtype 并输出。支持输出和输入的数据类型相同。

参数

  • x (Tensor) - 输入多维 Tensor,支持的数据类型为:bool、float16、float32、float64、uint8、int32、int64。

  • dtype (str|np.dtype) - 输出 Tensor 的数据类型。支持的数据类型为:bool、float16、float32、float64、int8、int32、int64、uint8。

返回

Tensor,维度与 x 相同,数据类型为 dtype

代码示例

import paddle

x = paddle.to_tensor([2, 3, 4], 'float64')
y = paddle.cast(x, 'uint8')

使用本API的教程文档