avoid big temporary tensor creation in matrix_to_axis_angle

Co-authored-by: Jeremy Reizenstein <[email protected]>

Author: alex-bene

pytorch3d/transforms/rotation_conversions.py

@@ -542,7 +542,7 @@ def matrix_to_axis_angle(matrix: torch.Tensor, fast: bool = False) -> torch.Tens
     zeros = torch.zeros(3, dtype=matrix.dtype, device=matrix.device)
     omegas = torch.where(torch.isclose(angles, torch.zeros_like(angles)), zeros, omegas)
 
-    near_pi = torch.isclose(angles, torch.full_like(angles, torch.pi)).squeeze(-1)
+    near_pi = angles.isclose(angles.new_tensor(torch.pi)).squeeze(-1)
 
     axis_angles = torch.empty_like(omegas)
     axis_angles[~near_pi] = (