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。

回傳類型:

Array