Warning

This page was created from a pull request.

jax.lax.reduce_window

jax.lax.reduce_window(operand, init_value, computation, window_dimensions, window_strides, padding, base_dilation=None, window_dilation=None)

Wraps XLA’s ReduceWindowWithGeneralPadding operator.

Parameters

• operand (Any) –

• init_value (Any) –

• computation (Callable) –

• window_dimensions (Sequence[int]) –

• window_strides (Sequence[int]) –

• padding (Union[str, Sequence[Tuple[int, int]]]) –

• base_dilation (Optional[Sequence[int]]) –

• window_dilation (Optional[Sequence[int]]) –

Return type

Any