Sum¶

class torchjd.aggregation.Sum[source]¶

Aggregator that sums of the rows of the input matrices.

__call__(matrix, /)[source]¶

Computes the aggregation from the input matrix and applies all registered hooks.

class torchjd.aggregation.SumWeighting[source]¶

Weighting that gives the weights \(\begin{bmatrix} 1 & \dots & 1 \end{bmatrix}^T \in \mathbb{R}^m\).

__call__(matrix, /)[source]¶

Computes the vector of weights from the input matrix and applies all registered hooks.

Parameters:: matrix (Tensor) – The matrix from which the weights must be extracted.
Return type:: Tensor