jax.numpy.isdtype — JAX documentation (original) (raw)

jax.numpy.isdtype#

jax.numpy.isdtype(dtype, kind)[source]#

Returns a boolean indicating whether a provided dtype is of a specified kind.

Parameters:

• dtype (DTypeLike) – the input dtype
• kind (str | DTypeLike | tuple_[_str | DTypeLike , ... ]) –
  the data type kind. If kind is dtype-like, return dtype = kind. If kind is a string, then return True if the dtype is in the specified category:
  • 'bool': {bool}
  • 'signed integer': {int4, int8, int16, int32, int64}
  • 'unsigned integer': {uint4, uint8, uint16, uint32, uint64}
  • 'integral': shorthand for ('signed integer', 'unsigned integer')
  • 'real floating': {float8_*, float16, bfloat16, float32, float64}
  • 'complex floating': {complex64, complex128}
  • 'numeric': shorthand for ('integral', 'real floating', 'complex floating')
    If kind is a tuple, then return True if dtype matches any entry of the tuple.

Returns:

True or False

Return type:

bool