
    
Ng                     6    d dl Zd dlmZ  G d de          ZdS )    N)OpRunc                   B    e Zd Zdej        dedej        fdZdddZdS )Concataaxisreturnc                     t          |j                  dk    rt          d|d          |t          |j                  k    r:|j        d|dz   t          |j                  z
  z  z   }|                    |          S |S )Nr   z&Concat: one input has an empty shape: .)   r   )lenshapeRuntimeErrorreshape)selfr   r   	new_shapes       X/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/reference/ops/op_concat.py_preprocesszConcat._preprocess   s{    qw<<1NNNNOOO3qw<<$$(S\\*A"BBI99Y'''    N)r   c                j     t           fd|D                       }t          j        |          fS )Nc              3   D   K   | ]}                     |          V  d S )N)r   ).0r   r   r   s     r   	<genexpr>zConcat._run.<locals>.<genexpr>   s3      >>Ad&&q$//>>>>>>r   )tuplenpconcatenate)r   r   argstargss   ``  r   _runzConcat._run   s>    >>>>>>>>>>ud++--r   )__name__
__module____qualname__r   ndarrayintr   r    r   r   r   r      s_        RZ s rz      $ . . . . . . .r   r   )numpyr   onnx.reference.op_runr   r   r$   r   r   <module>r'      sY        ' ' ' ' ' '. . . . .U . . . . .r   