Function homogenize_points

flowmap/model/projection.py:11–15

Convert batched points (xyz) to (xyz1).

(
    points: Float[Tensor, "*batch dim"],
)

def homogenize_points(
    points: Float[Tensor, "*batch dim"],
) -> Float[Tensor, "*batch dim+1"]:
    """Convert batched points (xyz) to (xyz1)."""
    return torch.cat([points, torch.ones_like(points[..., :1])], dim=-1)


def homogenize_vectors(