
    h                     .    S SK rS SKJr   " S S\5      rg)    N)OpRunUnaryNumc                       \ rS rSrSS jrSrg)Hardmax   Nc                     U=(       d    U R                   n[        R                  " XS9n[        R                  " U5      n[        R                  " U[        R
                  " X2S9SUS9  U4$ )N)axis   )r   npargmax
zeros_likeput_along_axisexpand_dims)selfxr   x_argmaxys        W/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/reference/ops/op_hardmax.py_runHardmax._run   sU     tyy99Q*MM!
r~~h2AD	
 t     )N)__name__
__module____qualname____firstlineno__r   __static_attributes__r   r   r   r   r      s    r   r   )numpyr
   onnx.reference.ops._opr   r   r   r   r   <module>r      s     0m r   