Pool¶

Helper function to define pooling functions.

Abstract Signature:

Pool(inputs: Tensor, init, reduce_fn: Callable, window_shape: Tuple, strides: Tuple, padding: str)

API: jax.lax.reduce_window

Strategy: Direct Mapping

API: numpy.random.bit_generator.SeedSequence.pool

Strategy: Direct Mapping

API: flax.nnx.pool

Strategy: Direct Mapping