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)

返回：

dtype 或 (dtype, weak_type)，取決於 return_weak_type 參數的值。

返回類型：

DType | tuple[DType, bool]