Skip to main content
Ctrl+K

jax.dtypes.issubdtype#

jax.dtypes.issubdtype(a, b)[source]#

Returns True if first argument is a typecode lower/equal in type hierarchy.

This is like numpy.issubdtype(), but can handle dtype extensions such as jax.dtypes.bfloat16 and jax.dtypes.prng_key.

Parameters:
  • a (Union[str, type[Any], dtype, SupportsDType, None]) –

  • b (Union[str, type[Any], dtype, SupportsDType, None]) –

Return type:

bool

previous

jax.dtypes.float0

next

jax.dtypes.prng_key