amend

pytorch · Oct 10, 2023 · db918a7 · db918a7
1 parent fb5c470
commit db918a7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tensordict/memmap.py b/tensordict/memmap.py
@@ -732,8 +732,7 @@ def to(
             return out
 
         out = self.clone()
-        out.device = device
-        return self
+        return out.to(device)
 
     def unbind(self, dim: int) -> tuple[torch.Tensor, ...]:
         """Unbinds a MemmapTensor along the desired dimension.