
    h                     B    S SK JrJrJr  S SKrS SKJr   " S S\5      rg)    )ListOptionalTupleN)OpRunc                       \ rS rSrS\R
                  S\\R
                     S\S\\R
                     4S jr	   SS\R
                  S\\R
                     S\S\S\
\R
                     4
S	 jjrS
rg)SplitToSequence   matsplitaxisreturnc                    Uc)  [        UR                  U   5       Vs/ s H  nSPM     nnOq[        UR                  5      S:X  aM  UR                  U   n[        U5      nU[        U5      -  nU/U-  nXgU-  -
  n	U	S:  a  UR	                  U	5        O[        U5      nUR                   V
s/ s H  n
[        SU
5      PM     nn
/ nSnU H4  n[        XU-   5      X'   X-  nUR	                  U[        U5         5        M6     U$ s  snf s  sn
f )N   r   )rangeshapelenintappendlistslicetuple)selfr
   r   r   _split_lengthdimlengthnleftssliresposspls                  a/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/reference/ops/op_split_to_sequence.py
common_runSplitToSequence.common_run   s    =',SYYt_'=>'=!A'=L>L"))D/CZFs6{"A"8a<L!#Dax##D);L$'II.IquQ{I.Cc9-CIJCJJs5:'   
) ? /s   D,D	Nkeepdimsc                     U R                  XUS9nUcY  U(       dR  [        [        U5      5       H:  n[        XV   R                  5      nXs	 XV   R                  [        U5      5      XV'   M<     U4$ )N)r   )r%   r   r   r   r   reshaper   )r   r
   r   r   r'   r!   ir   s           r$   _runSplitToSequence._run(   sc     oocto4=3s8_SV\\*Ke5 % v     )Nr   r   )__name__
__module____qualname____firstlineno__npndarrayr   r   r   r%   r   r+   __static_attributes__r.   r-   r$   r   r      s    ::&.rzz&:BE	bjj	: '+ZZ 

# 	
  
rzz	 r-   r   )	typingr   r   r   numpyr3   onnx.reference.op_runr   r   r.   r-   r$   <module>r9      s    ) (  '(e (r-   