U
    qhQ                     @  s4   d dl mZ d dlZd dlmZ G dd deZdS )    )annotationsN)Softmaxc                   @  s   e Zd Zdd ZdS )
LogSoftmaxc                 C  s$   t | |d }tj||d |fS )Nr   )out)r   _runnplog)selfXY r   E/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_log_softmax.pyr      s    zLogSoftmax._runN)__name__
__module____qualname__r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.reference.ops.op_softmaxr   r   r   r   r   r   <module>   s   