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

    Design a Mobile Site
    View Site in Mobile | Classic
    Share by: