jax.dtypes.result_type# jax.dtypes.result_type(*args, return_weak_type_flag=False)[源代码]# 方便的函数，用于应用 JAX 参数的类型提升。 参数: return_weak_type_flag (bool) – 如果为真，则返回一个 (dtype, weak_type) 元组。如果为假，则仅返回 dtype args (Any) 返回: 根据 return_weak_type 参数的值，返回 dtype 或 (dtype, weak_type)。 返回类型: DType | tuple[DType, bool]
