Constant

class torchjd.aggregation.Constant(weights)[source]

Aggregator that makes a linear combination of the rows of the provided matrix, with constant, pre-determined weights.

Parameters:

weights (Tensor) – The weights associated to the rows of the input matrices.

class torchjd.aggregation.ConstantWeighting(weights)[source]

Weighting that returns constant, pre-determined weights.

Parameters:

weights (Tensor) – The weights to return at each call.