Function homogenize_vectors

flowmap/model/projection.py:18–22

Convert batched vectors (xyz) to (xyz0).

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

def homogenize_vectors(
    vectors: Float[Tensor, "*batch dim"],
) -> Float[Tensor, "*batch dim+1"]:
    """Convert batched vectors (xyz) to (xyz0)."""
    return torch.cat([vectors, torch.zeros_like(vectors[..., :1])], dim=-1)


def transform_rigid(