diff --git a/norch/tensor.py b/norch/tensor.py index e660df9..8f92669 100644 --- 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,