
    h                     n    S SK rS SKJr  SS jrSS jr " S S\5      r " S S\5      r " S	 S
\5      rg)    N)OpRunc                     [         R                  " XS9nU(       aB  [        UR                  5      [        U R                  5      :  a  [         R                  " X15      nUR                  [         R                  5      $ )Naxis)npargmaxlenshapeexpand_dimsastypeint64datar   keepdimsresults       V/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/reference/ops/op_argmax.py_argmaxr      sL    YYt'FC%DJJ7-==""    c                     [         R                  " X5      n [         R                  " XS9nU R                  U   U-
  S-
  nU(       a  [         R                  " X15      nUR                  [         R                  5      $ )Nr      )r   flipr   r
   r   r   r   r   s       r   #_argmax_use_numpy_select_last_indexr      sZ    774DYYt'FZZ&*F-==""r   c                       \ rS rSrSS jrSrg)_ArgMax   Nc                     [        XUS94$ )Nr   r   )r   )selfr   r   r   s       r   _run_ArgMax._run   s    (;==r    )NN__name__
__module____qualname____firstlineno__r   __static_attributes__r!   r   r   r   r      s    >r   r   c                       \ rS rSrSrg)ArgMax_1    r!   N)r#   r$   r%   r&   r'   r!   r   r   r)   r)       s    r   r)   c                       \ rS rSrSS jrSrg)	ArgMax_12$   Nc                 L    US:X  a  [         R                  XX#S9$ [        XUS94$ )Nr   r   )r   r   r   )r   r   r   r   select_last_indexs        r   r   ArgMax_12._run%   s1    !<<<II/(S
 	
r   r!   )NNNr"   r!   r   r   r,   r,   $   s    
r   r,   )r   T)	numpyr   onnx.reference.op_runr   r   r   r   r)   r,   r!   r   r   <module>r3      s;     '##>e >
	w 	
 
r   