jax.numpy.promote_types

Contents

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 (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:

Returns:

A numpy.dtype object.