jax.lax.reduce_precision

jax.lax.reduce_precision(operand, exponent_bits, mantissa_bits)

包裝 XLA 的 ReducePrecision 運算子。

參數:

• operand (float | ArrayLike)

• exponent_bits (int)

• mantissa_bits (int)

回傳型別:

Array