tflite_support.task.processor.Embedding

Result produced by one of the embedder model output layers.

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

Creates a Embedding object from the given protobuf object.

to_pb2

View source

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

__eq__

View source

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.