fix: matmul bug in

2021-06-21 22:47:45 +02:00
parent cfe09ec06b
commit bc9a826b7d
2 changed files with 14 additions and 2 deletions
--- a/prototorch/core/transforms.py
+++ b/prototorch/core/transforms.py
@@ -36,7 +36,7 @@ class LinearTransform(torch.nn.Module):
        self._register_weights(weights)

    def forward(self, x):
-        return x @ self.weights.T
+        return x @ self.weights


 # Aliases