tflite_support.task.processor.EmbeddingResult

Embeddings produced by the Embedder.

tflite_support.task.processor.EmbeddingResult(
    embeddings: List[tflite_support.task.processor.Embedding]
)

embeddings The embeddings produced by each of the model output layers. Except in advanced cases, the embedding model has a single output layer, and this list is thus made of a single element feature vector.

Methods

create_from_pb2

View source

@classmethod
create_from_pb2(
    pb2_obj: _EmbeddingResultProto
) -> 'EmbeddingResult'

Creates a EmbeddingResult object from the given protobuf object.

to_pb2

View source

to_pb2() -> _EmbeddingResultProto

Generates a protobuf object to pass to the C++ layer.

__eq__

View source

__eq__(
    other: Any
) -> bool

Checks if this object is equal to the given object.

Args
other The object to be compared with.

Returns
True if the objects are equal.