jax.dtypes.canonicalize_dtype

jax.dtypes.canonicalize_dtype(dtype, allow_extended_dtype=False)

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

參數:

• dtype (Any)

• allow_extended_dtype (bool)

回傳類型:

DType | ExtendedDType