o
    èp©iÙ  ã                   @   s(   d dl Z ddlmZ G dd„ deƒZdS )é    Né   )ÚBaseReducerc                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
ÚMeanReducerc                 G   s
   t  |¡S ©N)ÚtorchÚmean)ÚselfÚlossesÚ_© r   úa/home/ubuntu/.local/lib/python3.10/site-packages/pytorch_metric_learning/reducers/mean_reducer.pyÚelement_reduction   s   
zMeanReducer.element_reductionc                 G   ó   | j |g|¢R Ž S r   ©r   ©r   r	   Úargsr   r   r   Úpos_pair_reduction
   ó   zMeanReducer.pos_pair_reductionc                 G   r   r   r   r   r   r   r   Úneg_pair_reduction   r   zMeanReducer.neg_pair_reductionc                 G   r   r   r   r   r   r   r   Útriplet_reduction   r   zMeanReducer.triplet_reductionN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r   r   r      s
    r   )r   Úbase_reducerr   r   r   r   r   r   Ú<module>   s    