
    wi                        d dl mZ 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 )    )annotationsN)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       l/root/.openclaw/workspace/chatterbox_venv_py311/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)
__future__r   numpyr   onnx.reference.op_runr   r   r   r   r&   r(   r$   r   r   <module>r.      s    # " " " " "     ' ' ' ' ' '# # # ## # # #> > > > >e > > >
	 	 	 	 	w 	 	 	
 
 
 
 
 
 
 
 
 
r   