
    wi                    B    d dl mZ d dlZd dlmZ  G d de          ZdS )    )annotationsN)OpRunc                  "    e Zd Zd
dZddd	ZdS )Concata
np.ndarrayaxisint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       l/root/.openclaw/workspace/chatterbox_venv_py311/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   )r   r   r	   r
   r   r   )__name__
__module____qualname__r   r!    r   r   r   r      sF             $ . . . . . . .r   r   )
__future__r   numpyr   onnx.reference.op_runr   r   r%   r   r   <module>r)      sk    # " " " " "     ' ' ' ' ' '. . . . .U . . . . .r   