
    Si4                    :    S SK Jr  S SKrS SKJr   " S S\5      rg)    )annotationsN)OpRunUnaryNumc                      \ rS rSrSS jrSrg)Hardmax   Nc                    U=(       d    U R                   nUR                  S:X  a  U4$ [        R                  " XS9n[        R                  " U5      n[        R
                  " U[        R                  " X2S9SUS9  U4$ )Nr   )axis   )r	   sizenpargmax
zeros_likeput_along_axisexpand_dims)selfxr	   x_argmaxys        \/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/reference/ops/op_hardmax.py_runHardmax._run   sj     tyy66Q;4K99Q*MM!
NN8/		
 t     )N)__name__
__module____qualname____firstlineno__r   __static_attributes__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   