jax.lax.reduce_window

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

Parameters:

• computation (Callable)

• window_dimensions (Sequence[int | Any])

• window_strides (Sequence[int])

• padding (str | Sequence[tuple[int, int]])

• base_dilation (Sequence[int] | None)

• window_dilation (Sequence[int] | None)

Return type:

Array