
    
Ng                     6    d dl Zd dlmZ  G d de          ZdS )    N)OpRunUnaryNumc                       e Zd ZddZdS )HardmaxNc                     |p| 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        Y/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/reference/ops/op_hardmax.py_runzHardmax._run   sl     ty9QT***M!
r~hT222AD	
 	
 	
 	
 t    )N)__name__
__module____qualname__r    r   r   r   r      s(             r   r   )numpyr	   onnx.reference.ops._opr   r   r   r   r   <module>r      sY        0 0 0 0 0 0    m     r   