
    
Ng                     ~    d dl Zd dlmZ ddZddZ G d de          Z G d d	e          Z G d
 de          ZdS )    N)OpRunTc                     t          j        | |          }|r?t          |j                  t          | j                  k     rt          j        ||          }|                    t           j                  S )Naxis)npargminlenshapeexpand_dimsastypeint64datar   keepdimsresults       X/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/reference/ops/op_argmin.py_argminr      s_    Yt$'''F .C%%DJ77--=="""    c                     t          j        | |          } t          j        | |          }| j        |         |z
  dz
  }|rt          j        ||          }|                    t           j                  S )Nr      )r   flipr   r
   r   r   r   r   s       r   #_argmin_use_numpy_select_last_indexr      sj    74DYt$'''FZ&*F .--=="""r   c                       e Zd ZddZdS )_ArgMinNc                 (    t          |||          fS )Nr   r   )r   )selfr   r   r   s       r   _runz_ArgMin._run   s    4(;;;==r   )NN__name__
__module____qualname__r    r   r   r   r      s(        > > > > > >r   r   c                       e Zd ZdS )ArgMin_1N)r    r!   r"   r#   r   r   r%   r%       s        Dr   r%   c                       e Zd ZddZdS )	ArgMin_12Nc                 p    |dk    rt                               | |||          S t          |||          fS )Nr   r   )r   r   r   )r   r   r   r   select_last_indexs        r   r   zArgMin_12._run%   sC    !!<<d<III/4(SSS
 	
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>r,      s        ' ' ' ' ' '# # # ## # # #> > > > >e > > >
	 	 	 	 	w 	 	 	
 
 
 
 
 
 
 
 
 
r   