Skip to content

pixano_inference.utils.vector

Utility functions for vector operations.

vector_to_tensor(vector)

Convert a vector to a tensor.

Parameters:

Name Type Description Default
vector NDArrayFloat | LanceVector | 'Tensor' | None

Vector to convert.

 required
Source code in pixano_inference/utils/vector.py 
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
def vector_to_tensor(vector: NDArrayFloat | LanceVector | "Tensor" | None) -> "Tensor":
    """Convert a vector to a tensor.

    Args:
        vector: Vector to convert.
    """
    assert_torch_installed()
    if not isinstance(vector, (NDArrayFloat, LanceVector, Tensor)) and vector is not None:
        raise ValueError(f"Unsupported vector type: {type(vector)}")

    if isinstance(vector, LanceVector):
        return vector.read_vector()
    elif isinstance(vector, NDArrayFloat):
        return vector.to_torch()
    return vector