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, returndtype = kind
. Ifkind
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: