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 | tuple[DType, bool]