
    h.                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   P    \ rS rSr\SS j5       r\SS j5       r\SS j5       rSrg)SplitToSequence   Nc                  6   [         R                  " S5      R                  S5      R                  [         R                  5      n [         R
                  " S[         R                  S9n[        R                  R                  SSS/S/S	S
9n[         R
                  " SS/SS/SS//[         R                  S9[         R
                  " SS/SS/SS//[         R                  S9[         R
                  " SS/SS/SS//[         R                  S9//n[        UX/USS9  g )N            dtyper   datasplitseq   axisg        g      ?g      @g      @g      (@g      *@g       @g      @g       @g      "@g      ,@g      .@g      @g      @g      $@g      &@g      0@g      1@test_split_to_sequence_1inputsoutputsnamenparangereshapeastypefloat32arrayint64onnxhelper	make_noder   r   r   nodeexpected_outputss       e/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/splittosequence.pyexport_with_split_1#SplitToSequence.export_with_split_1   s    yy}$$V,33BJJ?"((+{{$$05' % 
 3*sCj4,?rzzR3*sCj4,?rzzR3*tTlT4LAT
 	=$+		
    c                  H   [         R                  " S5      R                  S5      R                  [         R                  5      n [         R
                  " SS/[         R                  S9n[        R                  R                  SSS/S	/S
S9nU S S U SS  //n[        UX/USS9  g )Nr	   r
   r   r   r   r   r   r   r   r   r   test_split_to_sequence_2r   r   r&   s       r)   export_with_split_2#SplitToSequence.export_with_split_2%   s    yy}$$V,33BJJ?!Qrxx0{{$$05' % 
 RaQR
 	=$+		
r,   c                  R   [         R                  " S5      R                  S5      R                  [         R                  5      n [
        R                  R                  SS/S/SSS9n[        U R                  S   5       Vs/ s H  o S S 2U4   PM     sn/n[        UU /US	S
9  g s  snf )Nr	   r
   r   r   r   r   r   )r   keepdims!test_split_to_sequence_nokeepdimsr   )r   r   r   r   r    r#   r$   r%   rangeshaper   )r   r'   ir(   s       r)   export_nokeepdims!SplitToSequence.export_nokeepdims<   s    yy}$$V,33BJJ?{{$$HG % 
 27tzz!}1EF1EA!Q$Z1EFG6$4		
 Gs    B$ )returnN)	__name__
__module____qualname____firstlineno__staticmethodr*   r/   r7   __static_attributes__r9   r,   r)   r   r      s<    
 
. 
 
, 
 
r,   r   )numpyr   r#   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r9   r,   r)   <module>rD      s!   
   , .C
d C
r,   