jax.lax.reduce_window#
- jax.lax.reduce_window(operand, init_value, computation, window_dimensions, window_strides, padding, base_dilation=None, window_dilation=None)[source]#
Wraps XLA’s ReduceWindowWithGeneralPadding operator.
- Parameters:
computation (Callable)
window_dimensions (core.Shape)
window_strides (Sequence[int])
padding (str | Sequence[tuple[int, int]])
base_dilation (Sequence[int] | None)
window_dilation (Sequence[int] | None)
- Return type:
Array