jax.lax.bitcast_convert_type
jax.lax.bitcast_convert_type#
- jax.lax.bitcast_convert_type(operand, new_dtype)[source]#
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
- Return type
Array
- Returns
An array with the same shape as operand, bitcast elementwise to new_dtype.