jax.lax.convert_element_type

jax.lax.convert_element_type#

jax.lax.convert_element_type(operand, new_dtype)[source]#

逐元素转换。

包装 XLA 的 ConvertElementType 运算符，它执行从一种类型到另一种类型的逐元素转换。类似于 C++ static_cast。

参数:

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

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

返回:

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

返回类型:

数组

内容