ls_mlkit.util.vector_utils module

ls_mlkit.util.vector_utils.get_cosines_and_amplitudes(vectors: Tensor, mask: Tensor) → tuple[Tensor, Tensor]

ls_mlkit.util.vector_utils.get_vector_cosines(vectors: Tensor) → Tensor

Parameters:

vectors – torch.Tensor, shape (…, n_nodes, 3)

Returns:

torch.Tensor, shape (…, n_nodes * n_nodes)

Return type:

cosines