jax.lax.convert_element_type
jax.lax.convert_element_type#
- jax.lax.convert_element_type(operand, new_dtype)[source]#
Elementwise cast.
Wraps XLAβs ConvertElementType operator, which performs an elementwise conversion from one type to another. Similar to a C++ static_cast.