U
    qh                     @  sD   d dl mZ d dlZd dlmZ G dd deZG dd deZdS )    )annotationsN)OpRunReduceNumpyc                   @  s   e Zd ZdddZdS )ReduceLogSum_1NTc                 C  sl   |d k	rt |nd }|jdkr2| |tj ||S tj|||d}t|jdkr`tj||dfS t|fS Nr   )Zaxiskeepdims)out)	tuplesizereduce_constantnpinfsumlenshapelog)selfdataaxesr   Ztaxres r   H/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_reduce_log_sum.py_run   s    
zReduceLogSum_1._run)NT__name__
__module____qualname__r   r   r   r   r   r      s   r   c                   @  s   e Zd ZdddZdS )ReduceLogSum_18N   r   c                 C  s~   |  |r|r|fS | |}|dk}|jdkrD| |tj ||S tj|||d}t|jdkrrtj	||dfS t	|fS r   )
Zis_axes_emptyZhandle_axesr	   r
   r   r   r   r   r   r   )r   r   r   r   Znoop_with_empty_axesr   r   r   r   r      s    

zReduceLogSum_18._run)Nr   r   r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.reference.ops._opr   r   r   r   r   r   r   <module>   s   