jax.numpy.promote_types

jax.numpy.promote_types(a, b)

Returns the type to which a binary operation should cast its arguments.

For details of JAX’s type promotion semantics, see Type promotion semantics.

Parameters:

• a (Union[str, type[Any], dtype, SupportsDType]) – a numpy.dtype or a dtype specifier.

• b (Union[str, type[Any], dtype, SupportsDType]) – a numpy.dtype or a dtype specifier.

Return type:

dtype

Returns:

A numpy.dtype object.