
    /j                    v    d dl mZ d dlmZmZ d dlZd dlZd dlm	Z	 d dl
mZ erd dlmZ  G d de	          ZdS )	    )annotations)TYPE_CHECKINGAnyN)Base)expect)Sequencec                  &    e Zd Zedd            ZdS )ConcatreturnNonec                    ddgddgfddgddggddgddggfddgddggddgddgggd	d
gddggddgddgggfd} |                                  D ]\  }}d |D             }t          t          |d         j                            D ]}d t          t          |                    D             }t          j                            dt          |          dg|          }t          j	        ||          }t          |t          |          |gd|z   dz   t          |          z              t          t          |d         j                   d          D ]}d t          t          |                    D             }t          j                            dt          |          dg|          }t          j	        ||          }t          |t          |          |gd|z   dz   t          t          |                    z              d S )N                        	   
                     )1d2d3dc                N    g | ]"}t          j        |t           j                   #S ))dtype)npasarrayfloat32).0vs     g/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/concat.py
<listcomp>z!Concat.export.<locals>.<listcomp>   s)    GGG!bj"*555GGG    r   c                2    g | ]}d t          |          z   S valuestrr&   ks     r(   r)   z!Concat.export.<locals>.<listcomp>!   #    HHH7SVV+HHHr*   r
   output)inputsoutputsaxistest_concat__axis_)r4   r5   namec                2    g | ]}d t          |          z   S r,   r.   r0   s     r(   r)   z!Concat.export.<locals>.<listcomp>.   r2   r*   _axis_negative_)itemsrangelenshapeonnxhelper	make_nodelistr#   concatenater   r/   abs)
test_cases	test_casevalues_valuesiin_argsnoder3   s           r(   exportzConcat.export   st    q6Aq6"FQF#q!fq!f%56a&1a&!QFQF#34b'B8$Bx"b&:;0
 0

 #-"2"2"4"4 	 	IwGGwGGGF3vay//00  HHU3v;;5G5GHHH{,,T']]XJQ -   22<<#H')3h>QG	     Cq	000!44  HHU3v;;5G5GHHH{,,T']]XJQ -   22<<#H')36GG#cRSff++U	    	 	r*   N)r   r   )__name__
__module____qualname__staticmethodrM    r*   r(   r
   r
      s2        $ $ $ \$ $ $r*   r
   )
__future__r   typingr   r   numpyr#   r@   onnx.backend.test.case.baser   onnx.backend.test.case.noder   collections.abcr   r
   rR   r*   r(   <module>rY      s    # " " " " " % % % % % % % %      , , , , , , . . . . . . )((((((& & & & &T & & & & &r*   