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

Module: tfl.aggregation_layer

Layer which represents aggregation function.

See class level comment.

This layer applies the provided model to the ragged input tensor and aggregates the results.