tflite_support.task.processor.Embedding

Result produced by one of the embedder model output layers.

tflite_support.task.processor.Embedding(
    feature_vector: tflite_support.task.processor.FeatureVector,
    output_index: int
)

feature_vector The output feature vector.
output_index The index of the model output layer that produced this feature vector.

Methods

create_from_pb2

View source

@classmethod
create_from_pb2(
    pb2_obj: _EmbeddingProto
) -> 'Embedding'

Creates a Embedding object from the given protobuf object.

to_pb2

View source

to_pb2() -> _EmbeddingProto

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.