
    bIgD                    :    d dl mZ d dlZd dlmZ  G d de      Zy)    )annotationsN)OpRunUnaryNumc                      e Zd ZddZy)HardmaxNc                    |xs | j                   }t        j                  ||      }t        j                  |      }t        j                  |t        j
                  ||      d|       |fS )N)axis   )r   npargmax
zeros_likeput_along_axisexpand_dims)selfxr   x_argmaxys        Z/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/reference/ops/op_hardmax.py_runzHardmax._run   s\     tyy99QT*MM!
NN8$/		
 t    )N)__name__
__module____qualname__r    r   r   r   r      s    
r   r   )
__future__r   numpyr
   onnx.reference.ops._opr   r   r   r   r   <module>r      s    #  0m r   