U
    qh                     @  sT   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G dd	 d	eZdS )
    )annotationsN)OpRunc                   @  s   e Zd ZdddZdS )Unsqueeze_1Nc                 C  st   t |tjrt|}n$|g dfkr(d }nt |tr:t|}t |ttfrf|}|D ]}tj||d}qPntd|fS )N Zaxisz9axes cannot be None for operator Unsqueeze (Unsqueeze_1).)
isinstancenpZndarraytuplelistexpand_dimsRuntimeErrorselfdataZaxessqar   r   C/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_unsqueeze.py_run   s    

zUnsqueeze_1._run)N__name__
__module____qualname__r   r   r   r   r   r      s   r   c                   @  s   e Zd ZdS )Unsqueeze_11N)r   r   r   r   r   r   r   r      s   r   c                   @  s   e Zd ZdddZdS )Unsqueeze_13Nc                 C  s   |d k	rt |drt|jdkrztj|t|d}W q tk
r   t|dkrjtj|t|d d}n |}t|D ]}tj||d}qvY qX qtj||d}ntd|fS )N__iter__r   r      z:axes cannot be None for operator Unsqueeze (Unsqueeze_13).)	hasattrlenshaper   r   r	   	TypeErrorreversedr   r   r   r   r   r   #   s    zUnsqueeze_13._run)Nr   r   r   r   r   r   "   s   r   )	
__future__r   Znumpyr   Zonnx.reference.op_runr   r   r   r   r   r   r   r   <module>   s
   