jax.dtypes 模組#

bfloat16

bfloat16 浮點數值

canonicalize_dtype(dtype[, allow_extended_dtype])

根據 config.x64_enabled，將 dtype 從一種轉換為標準 dtype。

float0

對應於相同名稱的純量型別和 dtype 的 DType 類別。

issubdtype(a, b)

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

prng_key()

PRNG 金鑰 dtype 的純量類別。

result_type(*args[, return_weak_type_flag])

應用 JAX 引數 dtype 提升的便利函數。

scalar_type_of(x)

傳回與 JAX 值相關聯的純量型別。