From 46ff1c4eb124b8692e54987756f639a48ba63a27 Mon Sep 17 00:00:00 2001
From: Alexander Engelsberger <engelsbe@hs-mittweida.de>
Date: Tue, 29 Mar 2022 17:06:57 +0200
Subject: [PATCH] fix: forward of LinearTransform uses undetached weights now

---
 prototorch/core/transforms.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/prototorch/core/transforms.py b/prototorch/core/transforms.py
index 5e00bb0..04901cd 100644
--- 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
+        return x @ self._weights
 
 
 # Aliases