jax.dtypes
模块#
bfloat16 浮点数值 |
|
|
基于 config.x64_enabled,将数据类型转换为规范的数据类型。 |
与同名标量类型和数据类型对应的 DType 类。 |
|
|
如果第一个参数是类型层次结构中较低/等于的类型代码,则返回 True。 |
|
PRNG Key 数据类型的标量类。 |
|
应用 JAX 参数数据类型提升的便捷函数。 |
返回与 JAX 值关联的标量类型。 |
jax.dtypes
模块#bfloat16 浮点数值 |
|
|
基于 config.x64_enabled,将数据类型转换为规范的数据类型。 |
与同名标量类型和数据类型对应的 DType 类。 |
|
|
如果第一个参数是类型层次结构中较低/等于的类型代码,则返回 True。 |
|
PRNG Key 数据类型的标量类。 |
|
应用 JAX 参数数据类型提升的便捷函数。 |
返回与 JAX 值关联的标量类型。 |