jax.numpy.promote_types

jax.numpy.promote_types(a, b)[source]

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
Returns

A numpy.dtype object.