Upload pytorch3d_stub/pytorch3d/transforms/__init__.py with huggingface_hub

pytorch3d_stub/pytorch3d/transforms/__init__.py

@@ -102,6 +102,14 @@ class Transform3d:
         S[:, 0, 0] = s[:, 0]; S[:, 1, 1] = s[:, 1]; S[:, 2, 2] = s[:, 2]
         new_m = self._matrix @ S
         return Transform3d(matrix=new_m, device=self.device, dtype=self.dtype)
+    def to(self, device=None, dtype=None):
+        if device is not None: self.device = device
+        if dtype is not None: self.dtype = dtype
+        self._matrix = self._matrix.to(device=device, dtype=dtype)
+        return self
+    def inverse(self):
+        inv_m = torch.inverse(self._matrix)
+        return Transform3d(matrix=inv_m, device=self.device, dtype=self.dtype)
     def rotate(self, R):
         if R.dim() == 2: R = R.unsqueeze(0)
         T = torch.eye(4, dtype=self.dtype, device=self.device).unsqueeze(0).expand(R.shape[0], -1, -1).clone()