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
[Any
,str
,dtype
,SupportsDType
]) – anumpy.dtype
or a dtype specifier.b (
Union
[Any
,str
,dtype
,SupportsDType
]) – anumpy.dtype
or a dtype specifier.
- Return type:
- Returns:
A
numpy.dtype
object.