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.