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 – a numpy.dtype or a dtype specifier.

• b – a numpy.dtype or a dtype specifier.

Returns

A numpy.dtype object.