From d7a932f8edf95507083c8b39e294b608825b87d5 Mon Sep 17 00:00:00 2001 From: lucasdelimanogueira Date: Thu, 23 May 2024 09:48:16 -0300 Subject: [PATCH] fix send to device --- norch/tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,