
    h                     .    S SK rS SKJr   " S S\5      rg)    N)OpRunAiOnnxMlc                   N    \ rS rSr\S 5       r\S 5       r\S 5       rSS jrSr	g)	
Normalizer   c                     [         R                  " U 5      R                  SS9R                  U R                  S   S45      nU [         R
                  " US5      -  $ )zmax normalization   axisr   KH9)npabsmaxreshapeshapemaximumxdivs     c/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/reference/ops/aionnxml/op_normalizer.pynorm_maxNormalizer.norm_max   J     ffQimmm#++QWWQZ,<=2::c5)))    c                     [         R                  " U 5      R                  SS9R                  U R                  S   S45      nU [         R
                  " US5      -  $ )zL1 normalizationr   r	   r   r   r   )r   r   sumr   r   r   r   s     r   norm_l1Normalizer.norm_l1   r   r   c                     [         R                  " U 5      R                  SS9n[         R                  " XS9  [         R                  " UR                  U R                  S   S45      S5      nX-  $ )zL2 normalizationr   r	   )outr   r   r   )r   squarer   sqrtr   r   r   )r   xnnorms      r   norm_l2Normalizer.norm_l2   sX     YYq\1%
zz"**aggaj"%56>xr   Nc                     US:X  a  [         R                  nO=US:X  a  [         R                  nO&US:X  a  [         R                  nO[	        SU S35      eU" U5      4$ )NMAXL1L2zUnexpected value for norm='z'.)r   r   r   r%   
ValueError)selfr   r$   _norms       r   _runNormalizer._run    sY    5=''ET\&&ET\&&E:4&CDDa{r    )N)
__name__
__module____qualname____firstlineno__staticmethodr   r   r%   r.   __static_attributes__r0   r   r   r   r      sA    * *
 * *
  	r   r   )numpyr   ,onnx.reference.ops.aionnxml._op_run_aionnxmlr   r   r0   r   r   <module>r9      s     F r   