jax.dtypes.issubdtype#

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

如果第一个参数是类型层次结构中较低/相等的类型代码，则返回 True。

这类似于 numpy.issubdtype()，但可以处理 dtype 扩展，例如 jax.dtypes.bfloat16 和 jax.dtypes.prng_key。

参数:

  • a (DTypeLike | ExtendedDType | None)

  • b (DTypeLike | ExtendedDType | None)

返回类型:

bool