o
    wi                     @  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v rd }n	t |trt|}t |ttfr7|}t|D ]	}tj||d}q*|fS tj||d}|fS )N axis)
isinstancenpndarraytuplelistreversedsqueeze)selfdataaxessqar   r   Z/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/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 r   )r   __init__r   )r   	onnx_node
run_paramsr   r   r   r   !   s   
zSqueeze_13.__init__Nc                 C  sN   |d urt |drtj|t|d}|fS tj||d}|fS t|}|fS )N__iter__r   )hasattrr	   r   r   )r   r   r   r   r   r   r   r   %   s   

zSqueeze_13._runr   )r   r   r   r   r   r   r   r   r   r       s    r   )	
__future__r   numpyr	   onnx.reference.op_runr   r   r   r   r   r   r   r   <module>   s   