o iZ @sjddlZdedefddZ ddedBdedBd edefd d Z ddedBdedBd edefd d ZdS)N reductionreturncCs`|dkrd}|S|dkrd}|S|dkrtjdddd}|S|d kr'd}|Sd }t|d ) Nnonermeanelementwise_meanzPreduction='elementwise_mean' is deprecated. Please use reduction='mean' instead. stacklevelsumz# is not a valid value for reduction)warningswarn ValueError)rretrU/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/torch/nn/_reduction.pyget_enums$ rT size_averagereduce emit_warningcCsTd}|durd}|durd}|r|rd}n|rd}nd}|r(tj||dd|S)NzSsize_average and reduce args will be deprecated, please use reduction='{}' instead.Trr rrr )r rformat)rrrwarningrrrrlegacy_get_string srcCstt|||S)N)rr)rrrrrrlegacy_get_enum7sr)T)r strintrboolrrrrrrs,