
    bIgQ                    N    d dl mZ d dlZd dlZd dlmZ d dlmZ  G d de      Z	y)    )annotationsN)Base)expectc                  B    e Zd Zedd       Zedd       Zedd       Zy)SplitToSequencec                 R   t        j                  d      j                  d      j                  t         j                        } t        j
                  dt         j                        }t        j                  j                  dddgdgd	
      }t        j
                  ddgddgddggt         j                        t        j
                  ddgddgddggt         j                        t        j
                  ddgddgddggt         j                        gg}t        || |g|d       y )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       h/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/splittosequence.pyexport_with_split_1z#SplitToSequence.export_with_split_1   s    yy}$$V,33BJJ?"((+{{$$05' % 
 3*sCj4,?rzzR3*sCj4,?rzzR3*tTlT4LAT
 	%=$+		
    c                 X   t        j                  d      j                  d      j                  t         j                        } t        j
                  ddgt         j                        }t        j                  j                  dddgd	gd
      }| d d | dd  gg}t        || |g|d       y )Nr	   r
   r   r   r   r   r   r   r   r   r   test_split_to_sequence_2r   r   r&   s       r)   export_with_split_2z#SplitToSequence.export_with_split_2&   s    yy}$$V,33BJJ?!Qrxx0{{$$05' % 
 RaQR
 	%=$+		
r+   c                 X   t        j                  d      j                  d      j                  t         j                        } t
        j                  j                  ddgdgdd      }t        | j                  d         D cg c]  }| d d |f    c}g}t        || g|d	
       y c c}w )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_nokeepdimsz!SplitToSequence.export_nokeepdims=   s    yy}$$V,33BJJ?{{$$HG % 
 27tzz!}1EFAT!Q$ZFG6$4		
 Gs   B'N)returnNone)__name__
__module____qualname__staticmethodr*   r.   r5    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>rA      s$    #   , .C
d C
r+   