
    SiQ                    N    S SK Jr  S SKrS SKrS SKJr  S SKJr   " S S\5      r	g)    )annotationsN)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   c                 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       j/var/www/html/ai-image-ml/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$ N)returnNone)	__name__
__module____qualname____firstlineno__staticmethodr+   r0   r8   __static_attributes__r:   r-   r*   r   r      s<    
 
. 
 
, 
 
r-   r   )

__future__r   numpyr   r$   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r:   r-   r*   <module>rG      s$    #   , .C
d C
r-   