o
    èp©i  ã                   @   s    d dl mZ G dd„ deƒZdS )é   )ÚDotProductSimilarityc                       s   e Zd Z‡ fdd„Z‡  ZS )ÚCosineSimilarityc                    s.   t ƒ jdddi|¤Ž | jsJ ‚| jsJ ‚d S )NÚnormalize_embeddingsT© )ÚsuperÚ__init__Úis_invertedr   )ÚselfÚkwargs©Ú	__class__r   úg/home/ubuntu/.local/lib/python3.10/site-packages/pytorch_metric_learning/distances/cosine_similarity.pyr      s   
zCosineSimilarity.__init__)Ú__name__Ú
__module__Ú__qualname__r   Ú__classcell__r   r   r   r   r      s    r   N)Údot_product_similarityr   r   r   r   r   r   Ú<module>   s    