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
a – a
numpy.dtypeor a dtype specifier.b – a
numpy.dtypeor a dtype specifier.
- Returns
A
numpy.dtypeobject.