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 )	Squeeze_1Nc                 C  s~   t |tjrt|}n$|g dfkr(d }nt |tr:t|}t |ttfrj|}t|D ]}tj||d}qTntj||d}|fS )N Zaxis)
isinstancenpZndarraytuplelistreversedsqueeze)selfdataaxessqar   r   A/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_squeeze.py_run   s    

zSqueeze_1._run)N)__name__
__module____qualname__r   r   r   r   r   r      s   r   c                   @  s   e Zd ZdS )
Squeeze_11N)r   r   r   r   r   r   r   r      s   r   c                   @  s   e Zd Zdd ZdddZdS )
Squeeze_13c                 C  s   t | || d | _d S )N)r   __init__r   )r   Z	onnx_nodeZ
run_paramsr   r   r   r   !   s    zSqueeze_13.__init__Nc                 C  sF   |d k	r6t |dr&tj|t|d}q@tj||d}n
t|}|fS )N__iter__r   )hasattrr   r   r	   )r   r   r   r   r   r   r   r   %   s    

zSqueeze_13._run)N)r   r   r   r   r   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
   