jax.lax.reduce#

jax.lax.reduce(operands, init_values, computation, dimensions)[原始碼]#

包裝 XLA 的 Reduce 運算子。

init_values 和 computation 必須一起形成一個 么半群 以確保正確性。也就是說，init_values 必須是 computation 的單位元，且 computation 必須是結合律的。XLA 可能會在程式碼產生期間利用這兩個屬性；如果任一項被違反，結果將是未定義的。

參數:

  • operands (Any)

  • init_values (Any)

  • computation (Callable[[Any, Any], Any])

  • dimensions (Sequence[int])

返回類型:

Any