Skip to main content
Ctrl+K

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:

dtype

Returns:

A numpy.dtype object.

previous

jax.numpy.prod

next

jax.numpy.ptp