mp.tasks.components.containers.Embedding

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
)

embedding The actual embedding, either floating-point or scalar-quantized.
head_index The index of the embedder head that produced this embedding. This is useful for multi-head models.
head_name The name of the embedder head, which is the corresponding tensor metadata name (if any). This is useful for multi-head models.

Methods

__eq__

__eq__(
    other
)

head_index None
head_name None