jax.lax.convert_element_type#

jax.lax.convert_element_type(operand, new_dtype)[源代码]#

元素级类型转换。

封装 XLA 的 ConvertElementType 操作符，该操作符执行从一种类型到另一种类型的元素级转换。类似于 C++ 的 static_cast。

参数：

  • operand (ArrayLike) – 要转换的数组或标量值。

  • new_dtype (DTypeLike | dtypes.ExtendedDType) – 一个表示目标类型的 NumPy dtype。

返回:

一个与 operand 具有相同形状的数组，元素逐个转换为 new_dtype。

返回类型:

数组