jax.dtypes.issubdtype

目录

jax.dtypes.issubdtype#

jax.dtypes.issubdtype(a, b)[源代码][源代码]#

如果第一个参数在类型层次结构中低于或等于第二个参数，则返回 True。

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

参数:

  • a (DTypeLike | ExtendedDType | None)

  • b (DTypeLike | ExtendedDType | None)

返回类型:

bool

目录