amend

Author: vmoens

tensordict/base.py

@@ -4771,6 +4771,11 @@ def is_cuda(self):
     def is_cpu(self):
         return self.device is not None and self.device.type == "cpu"
 
+    @property
+    def is_meta(self):
+        dtype = self.dtype
+        return dtype is not None and self.dtype == torch.meta
+
     # Serialization functionality
     def state_dict(
         self,

tensordict/tensorclass.pyi

@@ -80,8 +80,6 @@ class TensorClass:
         lock: bool = False,
         **kwargs,
     ) -> None: ...
-    @property
-    def is_meta(self) -> bool: ...
     def __bool__(self) -> bool: ...
     def __ne__(self, other: object) -> T: ...
     def __xor__(self, other: TensorDictBase | float): ...
@@ -459,9 +457,11 @@ class TensorClass:
     def cpu(self, **kwargs) -> T: ...
     def cuda(self, device: int | None = None, **kwargs) -> T: ...
     @property
-    def is_cuda(self): ...
+    def is_cuda(self) -> bool: ...
     @property
-    def is_cpu(self): ...
+    def is_cpu(self) -> bool: ...
+    @property
+    def is_meta(self) -> bool: ...
     def state_dict(
         self,
         destination: Incomplete | None = None,