jax.dtypes.result_type

jax.dtypes.result_type(*args, return_weak_type_flag=False)

便捷函数，用于应用 JAX 参数 dtype 提升。

参数:

• return_weak_type_flag (bool) – 如果为 True，则返回一个 (dtype, weak_type) 元组。如果为 False，则只返回 dtype

• args (Any)

返回:

根据 return_weak_type 参数的值，返回 dtype 或 (dtype, weak_type)。

返回类型:

DType | tuple[DType, bool]