
    Ngx                     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 )ExpandreturnNc                     t           j                            dddgdg          } ddg}t          j        t          j        dt          j        |          dz   t          j                  |          }g d	}|t          j        |t          j                  z  }t          j	        |t          j
                  }t          | ||g|gd
           d S )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        ^/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/expand.pyexport_dim_changedzExpand.export_dim_changed   s    {$$K(L % 
 

 Az")Arwu~~'9LLLeTT II	"')2:>>>> HYbh777	)$J*		
 	
 	
 	
 	
 	
    c                     t           j                            dddgdg          } ddg}ddg}t          j        t          j        dt          j        |          dz   t          j        	          |          }t          j        |d          }t          j	        |t          j
        	          }t          | ||g|gd
           d S )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_unchangedzExpand.export_dim_unchanged*   s    {$$K(L % 
 

 AF	z")Arwu~~'9LLLeTT 74##
 HYbh777	)$J,		
 	
 	
 	
 	
 	
r'   )r   N)__name__
__module____qualname__staticmethodr&   r,    r'   r%   r   r      sR        
 
 
 \
8 
 
 
 \
 
 
r'   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r1   r'   r%   <module>r5      st   
      , , , , , , . . . . . .5
 5
 5
 5
 5
T 5
 5
 5
 5
 5
r'   