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
Return type

Any