import torch import numpy as np from keymorph.keypoint_aligners import RigidKeypointAligner bs, num_points = 8, 10 points_m = torch.rand(bs, num_points, 3) points_f = torch.rand(bs, num_points, 3) ...