lgatr.interface.vector

Embedding and extracting vectors into multivectors.

Functions

embed_vector(vector)	Embeds Lorentz vectors in multivectors.
extract_vector(multivector)	Given a multivector, extract a Lorentz vector.

lgatr.interface.vector.embed_vector(vector)

Embeds Lorentz vectors in multivectors.

Parameters:

vector (torch.Tensor) – Lorentz vector with shape (…, 4)

Returns:

multivector – Embedding into multivector with shape (…, 16).

Return type:

torch.Tensor

lgatr.interface.vector.extract_vector(multivector)

Given a multivector, extract a Lorentz vector.

Parameters:

multivector (torch.Tensor) – Multivector with shape (…, 16).

Returns:

vector – Lorentz vector with shape (…, 4)

Return type:

torch.Tensor