jax.dtypes 模块#

bfloat16

bfloat16 浮点数值

canonicalize_dtype(dtype[, allow_extended_dtype])

基于 config.x64_enabled,将数据类型转换为规范的数据类型。

float0

与同名标量类型和数据类型对应的 DType 类。

issubdtype(a, b)

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

prng_key()

PRNG Key 数据类型的标量类。

result_type(*args[, return_weak_type_flag])

应用 JAX 参数数据类型提升的便捷函数。

scalar_type_of(x)

返回与 JAX 值关联的标量类型。