jax.dtypes.result_type

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

应用 JAX 参数类型提升的便捷函数。

参数:

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

• args (Any)

返回:

dtype 或 (dtype, weak_type) 取决于 return_weak_type 参数的值。

返回类型:

DType | 元组[DType, 布尔值]