Update wilor/utils/geometry.py

wilor/utils/geometry.py

@@ -58,7 +58,7 @@ def rot6d_to_rotmat(x: torch.Tensor) -> torch.Tensor:
     a2 = x[:, :, 1]
     b1 = F.normalize(a1)
     b2 = F.normalize(a2 - torch.einsum('bi,bi->b', b1, a2).unsqueeze(-1) * b1)
-    b3 = torch.cross(b1, b2)
+    b3 = torch.linalg.cross(b1,b2) #torch.cross(b1, b2)
     return torch.stack((b1, b2, b3), dim=-1)
 
 def perspective_projection(points: torch.Tensor,