
    Ng                     J    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 )    N)Base)expectc                   >    e Zd Zedd            Zedd            ZdS )SqueezereturnNc                  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       _/var/www/html/ai-engine/env/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*   )r   N)__name__
__module____qualname__staticmethodr)   r.    r*   r(   r   r      s\        
I 
I 
I \
I 	W 	W 	W \	W 	W 	Wr*   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r3   r*   r(   <module>r7      s~   
      , , , , , , . . . . . .W W W W Wd W W W W Wr*   