fix send to device

2024-05-23 09:48:16 -03:00 · 2024-05-23 09:48:16 -03:00 · d7a932f8ed
commit d7a932f8ed
parent 065e913b2d
1 changed files with 1 additions and 1 deletions
--- a/norch/tensor.py
+++ b/norch/tensor.py
@ -53,7 +53,7 @@ class Tensor:
                self.device_ctype
            )

-            self.to(device)
+            self = self.to(device)
        
        else:
            self.tensor = None,