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 | Tensor | None

Vector to convert.

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

    Args:
        vector: Vector to convert.
    """
    from torch import Tensor

    from pixano_inference.schemas.nd_array import NDArrayFloat

    assert_torch_installed()
    if not isinstance(vector, (NDArrayFloat, Tensor)) and vector is not None:
        raise ValueError(f"Unsupported vector type: {type(vector)}")

    if isinstance(vector, NDArrayFloat):
        return vector.to_torch()
    return vector