Aggregator (abstract)

class torchjd.aggregation.bases.Aggregator(*args, **kwargs)

Bases: Module, ABC

Abstract base class for all aggregators. It has the role of aggregating matrices of dimension \(m \times n\) into row vectors of dimension \(n\).