jax.lax.reduce_precision

jax.lax.reduce_precision#

jax.lax.reduce_precision(operand, exponent_bits, mantissa_bits)[source]#

包装 XLA 的 ReducePrecision 运算符。

参数:
  • operand (float | ArrayLike)

  • exponent_bits (int)

  • mantissa_bits (int)

返回类型:

数组