jax.dtypes.canonicalize_dtype

jax.dtypes.canonicalize_dtype#

jax.dtypes.canonicalize_dtype(dtype, allow_extended_dtype=False)[源代码]#

基于 config.x64_enabled 将数据类型转换为规范数据类型。

参数:
  • dtype (Any)

  • allow_extended_dtype (bool)

返回类型:

DType | ExtendedDType