![]() |
Embedding result for a given embedder head.
mp.tasks.components.containers.Embedding(
embedding: np.ndarray,
head_index: Optional[int] = None,
head_name: Optional[str] = None
)
Methods
create_from_pb2
@classmethod
create_from_pb2( pb2_obj:
mp.tasks.components.utils.cosine_similarity.embedding_result.embeddings_pb2.Embedding
) -> 'Embedding'
Creates a Embedding
object from the given protobuf object.
__eq__
__eq__(
other
)
Class Variables | |
---|---|
head_index |
None
|
head_name |
None
|