lgatr.interface.axialvector

Embedding and extracting axial vectors into multivectors.

Functions

embed_axialvector(axialvector)

Embeds axial vectors in multivectors.

extract_axialvector(multivector)

Given a multivector, extract a axial vector.

lgatr.interface.axialvector.embed_axialvector(axialvector)[source]

Embeds axial vectors in multivectors.

Parameters:

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

Returns:

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

Return type:

torch.Tensor

lgatr.interface.axialvector.extract_axialvector(multivector)[source]

Given a multivector, extract a axial vector.

Parameters:

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

Returns:

axialvector – Axial vector with shape (…, 4)

Return type:

torch.Tensor