From 0e435c297c95c9cd949fcf5ecadb00fd78fc72d2 Mon Sep 17 00:00:00 2001
From: generatedunixname1417043136753450 <noreply+1417043136753450@fb.com>
Date: Mon, 23 Feb 2026 12:28:37 -0800
Subject: [PATCH] 
 fbcode/vision/fair/pytorch3d/pytorch3d/ops/points_alignment.py

Reviewed By: sgrigory

Differential Revision: D93712744

fbshipit-source-id: 660560cdef9ff1d2173ae06de54df31766ee537f
---
 pytorch3d/ops/points_alignment.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/pytorch3d/ops/points_alignment.py b/pytorch3d/ops/points_alignment.py
index 96e4b410..6ca2950e 100644
--- a/pytorch3d/ops/points_alignment.py
+++ b/pytorch3d/ops/points_alignment.py
@@ -182,8 +182,7 @@ def iterative_closest_point(
         t_history.append(SimilarityTransform(R, T, s))
 
         # compute the root mean squared error
-        # pyre-fixme[58]: `**` is not supported for operand types `Tensor` and `int`.
-        Xt_sq_diff = ((Xt - Xt_nn_points) ** 2).sum(2)
+        Xt_sq_diff = torch.square(Xt - Xt_nn_points).sum(2)
         rmse = oputil.wmean(Xt_sq_diff[:, :, None], mask_X).sqrt()[:, 0, 0]
 
         # compute the relative rmse