o
    toiw                     @  sL   d dl mZ d dlmZ d dlZd dlmZ ddddZG dd deZ	dS )    )annotations)AnyN)OpRunseq	list[Any]axisintnew_axisreturn
np.ndarrayc                   sf   |dkr* dkrdd | D }t j|dd}|S  fdd| D }t j| d}|S t j|  d}|S )N   c                 S  s   g | ]	}|d t jf qS ).)npnewaxis.0s r   ^/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_concat_from_sequence.py
<listcomp>   s    z)_concat_from_sequence.<locals>.<listcomp>r   c                   s   g | ]}t | qS r   )r   expand_dimsr   r   r   r   r      s    )r   concatenate)r   r   r	   seq2resr   r   r   _concat_from_sequence   s   r   c                   @  s   e Zd ZdddZdS )ConcatFromSequenceNc                 C  s$   |d u rt dt|||d}|fS )NzA sequence cannot be null.)r	   )RuntimeErrorr   )selfr   r   r	   r   r   r   r   _run   s   zConcatFromSequence._run)NN)__name__
__module____qualname__r   r   r   r   r   r      s    r   )r   )r   r   r   r   r	   r   r
   r   )

__future__r   typingr   numpyr   onnx.reference.op_runr   r   r   r   r   r   r   <module>   s   