
    /j                    Z    d dl mZ d dlmZ d dlZd dlmZ 	 dddZ G d de          Z	dS )    )annotations)AnyN)OpRunsequencelist[Any] | np.ndarraytensor
np.ndarraypositionnp.ndarray | Nonereturn	list[Any]c                   g }| t          | t          j                  rt          | j                  dk    rM	 |                    |            n6# t          $ r)}t          dt          |            d|  d          |d }~ww xY w|?|d         t          |          z   t          |          z  }|                    ||           n|	                    |           |S )Nr   zUnable to iterate on type z: .)

isinstancenpndarraylenshapeextend	TypeErrortypeinsertappend)r   r   r
   seqeinsert_positions         j/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/reference/ops/op_sequence_insert.py(sequence_insert_reference_implementationr      s     Cx,, 03HN0C0Ca0G0G	JJx     	 	 	JT(^^JJxJJJ 	  $A;S1SXX=

?F++++ 	

6Js   A 
B$A<<Bc                      e Zd ZddZdS )SequenceInsertNc                b   |t          ||          }nt          |t                    rt          |||g          }nrt          |j                  dk    rt          |||          }nHt          |j                  dk    r t          ||t          |          g          }nt          ||          }|fS )Nr   )r   r   intr   r   )selfSTindress        r   _runzSequenceInsert._run)   s    ;:1a@@CCS!! 	A:1a#GGCC^^a:1aEECC^^q  :1a#c((LLCC:1a@@Cv    N)__name__
__module____qualname__r(    r)   r   r    r    (   s(             r)   r    r*   )r   r   r   r	   r
   r   r   r   )

__future__r   typingr   numpyr   onnx.reference.op_runr   r   r    r.   r)   r   <module>r3      s    # " " " " "           ' ' ' ' ' ' #'    6    U     r)   