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