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)OpRunc                   @  s   e Zd ZddddZdS )EinsumN)equationc                G  sx   t |tstdt|d| }|s2tdztj|f|ddifW S  tk
rr   tj|f| f Y S X d S )Nzequation must be string but is .zequation is empty.optimizeT)
isinstancestr	TypeErrortypestripnpZeinsum)selfr   args r   @/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_einsum.py_run   s    
zEinsum._run)__name__
__module____qualname__r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.reference.op_runr   r   r   r   r   r   <module>   s   