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