U
    ‰d  ã                   @   sd   d dl mZ d dlZeedœdd„Zdee ee eedœdd	„Zdee ee eedœd
d„ZdS )é    )ÚOptionalN)Ú	reductionÚreturnc                 C   sX   | dkrd}nF| dkrd}n8| dkr4t  d¡ d}n | dkrBd}nd	}td
 | ¡ƒ‚|S )NÚnoner   Úmeané   Zelementwise_meanzPreduction='elementwise_mean' is deprecated, please use reduction='mean' instead.Úsumé   éÿÿÿÿz%{} is not a valid value for reduction)ÚwarningsÚwarnÚ
ValueErrorÚformat)r   Úret© r   ú7/tmp/pip-unpacked-wheel-ua33x9lu/torch/nn/_reduction.pyÚget_enum   s    
r   T)Úsize_averageÚreduceÚemit_warningr   c                 C   sP   d}| d krd} |d krd}| r*|r*d}n|r4d}nd}|rLt  | |¡¡ |S )NzSsize_average and reduce args will be deprecated, please use reduction='{}' instead.Tr   r   r   )r   r   r   )r   r   r   Úwarningr   r   r   r   Úlegacy_get_string   s    r   c                 C   s   t t| ||ƒƒS )N)r   r   )r   r   r   r   r   r   Úlegacy_get_enum.   s    r   )T)T)	Útypingr   r   ÚstrÚintr   Úboolr   r   r   r   r   r   Ú<module>   s   