jax.dtypes.issubdtype#

jax.dtypes.issubdtype(a, b)[source]#

如果第一個參數是型別階層中較低/相等的型別碼，則傳回 True。

這類似於 numpy.issubdtype()，但可以處理 dtype 擴充功能，例如 jax.dtypes.bfloat16 和 jax.dtypes.prng_key。

參數:

  • a (DTypeLike | ExtendedDType | None)

  • b (DTypeLike | ExtendedDType | None)

傳回型別:

bool