
    Si                    N    S SK Jr  S SKrS SKrS SKJr  S SKJr   " S S\5      r	g)    )annotationsN)Base)expectc                  <    \ rS rSr\SS j5       r\SS j5       rSrg)Expand   c                    [         R                  R                  SSS/S/S9n SS/n[        R                  " [        R
                  " S[        R                  " U5      S-   [        R                  S9U5      n/ S	QnU[        R                  " U[        R                  S9-  n[        R                  " U[        R                  S9n[        U X#/U/S
S9  g )Nr   data	new_shapeexpandedinputsoutputs      dtype)   r      test_expand_dim_changedr   r   name)onnxhelper	make_nodenpreshapearangeprodfloat32onesarrayint64r   )nodeshaper
   r   r   s        a/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/expand.pyexport_dim_changedExpand.export_dim_changed   s    {{$$K(L % 

 Azz"))Arwwu~'9LeT 	"'')2::>> HHYbhh7	$J*		
    c                    [         R                  R                  SSS/S/S9n SS/nSS/n[        R                  " [        R
                  " S[        R                  " U5      S-   [        R                  S	9U5      n[        R                  " US5      n[        R                  " U[        R                  S	9n[        U X2/U/S
S9  g )Nr   r
   r   r   r   r   r      r   test_expand_dim_unchangedr   )r   r   r   r   r   r   r   r    tiler"   r#   r   )r$   r%   r   r
   r   s        r&   export_dim_unchangedExpand.export_dim_unchanged+   s    {{$$K(L % 

 AF	zz"))Arwwu~'9LeT 774#
 HHYbhh7	$J,		
r)    N)returnNone)__name__
__module____qualname____firstlineno__staticmethodr'   r.   __static_attributes__r0   r)   r&   r   r      s(    
 
8 
 
r)   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r0   r)   r&   <module>r=      s"    #   , .5
T 5
r)   