
    bIg                    v    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
y)    )annotationsN)OpRunc                    t        j                  | |      }|rAt        |j                        t        | j                        k  rt        j                  ||      }|j                  t         j                        S )Naxis)npargminlenshapeexpand_dimsastypeint64datar   keepdimsresults       Y/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/reference/ops/op_argmin.py_argminr      sP    YYt$'FC%DJJ7-==""    c                    t        j                  | |      } t        j                  | |      }| j                  |   |z
  dz
  }|rt        j                  ||      }|j                  t         j                        S )Nr      )r   flipr	   r   r   r   r   r   s       r   #_argmin_use_numpy_select_last_indexr      s`    774DYYt$'FZZ&*F-==""r   c                      e Zd ZddZy)_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y)ArgMin_1N)r!   r"   r#   r$   r   r   r&   r&       s    r   r&   c                      e Zd ZddZy)	ArgMin_12Nc                \    |dk(  rt         j                  | |||      S t        |||      fS )Nr   r   )r   r   r   )r   r   r   r   select_last_indexs        r   r   zArgMin_12._run%   s7    !<<d<II/4(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   