U
    qh                     @  s4   d dl mZ d dlZd dlmZ G dd deZdS )    )annotationsN)OpRunUnaryNumc                   @  s   e Zd ZdddZdS )SoftmaxNc                 C  sF   |p| j }||j|dd }t|}||j|dd }||jfS )N   )axisZkeepdims)r   maxnpexpsumZastypeZdtype)selfXr   tmpY r   A/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_softmax.py_run   s
    

zSoftmax._run)N)__name__
__module____qualname__r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.reference.ops._opr   r   r   r   r   r   <module>   s   