
    /j?                    V    d dl mZ d dlZd dlZd dlmZ d dlmZ  G d de          Z	dS )    )annotationsN)Base)expectc                  >    e Zd Zedd            Zedd            ZdS )SqueezereturnNonec                 h   t           j                            dddgdg          } t          j                            dddd	                              t          j                  }t          j        d
gt          j	                  }t          j
        |d
          }t          | ||g|gd           d S )Nr   xaxesyinputsoutputs            r   dtypeaxistest_squeezer   r   nameonnxhelper	make_nodenprandomrandnastypefloat32arrayint64squeezer   noder   r   r   s       h/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/squeeze.pyexport_squeezezSqueeze.export_squeeze   s    {$$=E % 
 

 IOOAq!Q''..rz::x28,,,Jqq!!!tQIsHHHHHH    c                 h   t           j                            dddgdg          } t          j                            dddd                              t          j                  }t          j        d	gt          j	        
          }t          j
        |d	          }t          | ||g|gd           d S )Nr   r   r   r   r   r   r   r   r   r   test_squeeze_negative_axesr   r   r(   s       r*   export_squeeze_negative_axesz$Squeeze.export_squeeze_negative_axes   s    {$$=E % 
 

 IOOAq!Q''..rz::xBH---Jqr"""tQIs9UVVVVVVr,   N)r   r	   )__name__
__module____qualname__staticmethodr+   r0    r,   r*   r   r      s\        
I 
I 
I \
I 	W 	W 	W \	W 	W 	Wr,   r   )

__future__r   numpyr    r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r5   r,   r*   <module>r:      s    # " " " " "      , , , , , , . . . . . .W W W W Wd W W W W Wr,   