Matrix

class torchjd.linalg.Matrix

Tensor with exactly 2 dimensions.

Common examples include the Jacobian matrix J of shape [m, n], where m is the number of objectives and n is the number of model parameters, and the Gramian of the Jacobian G = J J^T of shape [m, m].

Note

This class should never be instantiated. It is only used for static type checking.