
    bIg?                    N    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	y)    )annotationsN)Base)expectc                  0    e Zd Zedd       Zedd       Zy)Squeezec                 j   t         j                  j                  dddgdg      } t        j                  j                  dddd	      j                  t        j                        }t        j                  d
gt        j                        }t        j                  |d
      }t        | ||g|gd       y )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/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/squeeze.pyexport_squeezezSqueeze.export_squeeze   s    {{$$=E % 

 IIOOAq!Q'..rzz:xx288,JJqq!tQIsH    c                 j   t         j                  j                  dddgdg      } t        j                  j                  dddd      j                  t        j                        }t        j                  d	gt        j                  
      }t        j                  |d	      }t        | ||g|gd       y )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 % 

 IIOOAq!Q'..rzz:xxBHH-JJqr"tQIs9UVr*   N)returnNone)__name__
__module____qualname__staticmethodr)   r.    r*   r(   r   r      s,    
I 
I 	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d Wr*   