ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: oryx.experimental.nn.pooling

Contains building blocks for pooling layers used for neural networks.

Classes

class AvgPooling: Average pooling layer, computes the average within the window.

class MaxPooling: Max pooling layer, computes the maximum within the window.

class Pooling: Base class for Pooling layers.

class SumPooling: Sum pooling layer, computes the sum within the window.