Ingmar Bergman
Results
1
comments of
Ingmar Bergman
Try to replace 'input = input.unsqueeze(-2).expand(*input.shape[:-(len(input.shape) - 2)], self.out_features, self.in_features)' by 'input = input.unsqueeze(-2).expand(*input.shape[:-1], self.out_features, self.in_features)'. It worked for me.