jax.dtypes.issubdtype

内容

jax.dtypes.issubdtype#

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

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

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

参数：:

  • a (DTypeLike | ExtendedDType | None)

  • b (DTypeLike | ExtendedDType | None)

返回值：:

bool

内容