refact

2024-09-09 10:50:54 +02:00
parent 9d9c2aa23e
commit f0d68b34c7
23 changed files with 480 additions and 229 deletions
--- a/pina/label_tensor.py
+++ b/pina/label_tensor.py
@@ -427,4 +427,8 @@ class LabelTensor(torch.Tensor):
    def requires_grad_(self, mode=True):
        lt = super().requires_grad_(mode)
        lt.labels = self.labels
-        return lt
+        return lt
+
+    @property
+    def dtype(self):
+        return super().dtype