jax.lax.bitcast_convert_type

jax.lax.bitcast_convert_type(operand, new_dtype)

Elementwise bitcast.

Wraps XLA’s BitcastConvertType operator, which performs a bit cast from one type to another. The bitwidth of the source and destination types must match.

Parameters

• operand (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – an array or scalar value to be cast

• new_dtype (Union[Any, str, dtype, SupportsDType]) – the new type. Should be a NumPy type.

Return type

Array

Returns

An array with the same shape as operand, bitcast elementwise to new_dtype.