Aggregator (abstract)

class torchjd.aggregation.Aggregator(*args, **kwargs)[source]

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\).