Method A

cdslib/core/nn/modules/subspace.py:133–136

(self, x)

        return weight

    def A(self, x):
        weight = self._get_weight()
        y = F.linear(input=x, weight=weight, bias=None)
        return y, weight

    def AT(self, y):
        weight = self._get_weight()