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.Ifkind is dtype-like, returndtype=kind.Ifkind is a string, then return True if the dtype is in the specified category:

  • 'bool':{bool}

  • 'signedinteger':{int4,int8,int16,int32,int64}

  • 'unsignedinteger':{uint4,uint8,uint16,uint32,uint64}

  • 'integral': shorthand for('signedinteger','unsignedinteger')

  • 'realfloating':{float8_*,float16,bfloat16,float32,float64}

  • 'complexfloating':{complex64,complex128}

  • 'numeric': shorthand for('integral','realfloating','complexfloating')

  Ifkind is a tuple, then return True if dtype matches any entry of the tuple.

Returns:

True or False

Return type:

bool