jax.lax.reduce_window

Warning

This page was created from a pull request (#9655).

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 (Sequence[Union[int, Any]]) –
window_strides (Sequence[int]) –
padding (Union[str, Sequence[Tuple[int, int]]]) –
base_dilation (Optional[Sequence[int]]) –
window_dilation (Optional[Sequence[int]]) –

Return type

previous

jax.lax.reduce_precision

next

jax.lax.reshape