jax.dtypes.issubdtype#

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

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

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

参数:

  • a (DTypeLike | ExtendedDType | None)

  • b (DTypeLike | ExtendedDType | None)

返回类型:

bool