jax.dtypes.canonicalize_dtype# jax.dtypes.canonicalize_dtype(dtype, allow_extended_dtype=False)[源代码]# 基于 config.x64_enabled 将数据类型转换为规范数据类型。 参数: dtype (Any) allow_extended_dtype (bool) 返回类型: DType | ExtendedDType