
    wi4                    B    d dl mZ d dlZd dlmZ  G d de          ZdS )    )annotationsN)OpRunUnaryNumc                      e Zd ZddZdS )HardmaxNc                    |p| j         }|j        dk    r|fS t          j        ||          }t          j        |          }t          j        |t          j        ||          d|           |fS )Nr   )axis   )r   sizenpargmax
zeros_likeput_along_axisexpand_dims)selfxr   x_argmaxys        m/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/onnx/reference/ops/op_hardmax.py_runzHardmax._run   s     ty6Q;;4K9QT***M!
N8$///		
 	
 	
 	
 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      sk    # " " " " "     0 0 0 0 0 0    m     r   