
    wi                    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 )SoftmaxNc                    |j         dk    r|fS |p| j        }||                    |d          z
  }t          j        |          }||                    |d          z  }|                    |j                  fS )Nr      )axiskeepdims)sizer	   maxnpexpsumastypedtype)selfXr	   tmpYs        m/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/onnx/reference/ops/op_softmax.py_runzSoftmax._run   sx    6Q;;4K ty!%%TA%...F3KK	QUUqU)))!!##    )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   