Random

class torchjd.scalarization.Random

Scalarizer that combines the input tensor of values with positive random weights summing to 1, as defined in Algorithm 2 of Reasonable Effectiveness of Random Weighting: A Litmus Test for Multi-Task Learning.

__call__(values, /)

Computes the scalar value from the input tensor of values and applies all registered hooks.

Parameters:

values (Tensor) – The tensor of values to scalarize. May be of any shape.

Return type:

Tensor