jax.dtypes.canonicalize_dtype

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

根据 config.x64_enabled 将数据类型转换为规范的数据类型。

参数：

• dtype (Any)

• allow_extended_dtype (bool)

返回类型：

DType | ExtendedDType