
    /jQ                    V    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	dS )    )annotationsN)Base)expectc                  V    e Zd Zedd            Zedd            Zedd            ZdS )SplitToSequencereturnNonec                 :   t          j        d                              d                              t           j                  } t          j        dt           j                  }t          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           d S )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       p/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/splittosequence.pyexport_with_split_1z#SplitToSequence.export_with_split_1   s&   y}}$$V,,33BJ??"(+++{$$05' % 
 
 3*sCj4,?rzRRR3*sCj4,?rzRRR3*tTlT4LATTT
 	%=$+		
 	
 	
 	
 	
 	
    c                 r   t          j        d                              d                              t           j                  } t          j        ddgt           j                  }t          j        	                    dddgd	gd
          }| d d         | dd          gg}t          || |g|d           d S )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    y}}$$V,,33BJ??!Qrx000{$$05' % 
 
 RaRQRR
 	%=$+		
 	
 	
 	
 	
 	
r-   c                 R   t          j        d                              d                              t           j                  t
          j                            ddgdgdd          } fd	t          j	        d                   D             g}t          | g|d
           d S )Nr   r   r   r   r   r   r   )r   keepdimsc                (    g | ]}d d |f         S )N ).0ir   s     r+   
<listcomp>z5SplitToSequence.export_nokeepdims.<locals>.<listcomp>I   s%    FFFAT!!!Q$ZFFFr-   !test_split_to_sequence_nokeepdimsr   )r   r   r    r!   r"   r%   r&   r'   rangeshaper   )r)   r*   r   s     @r+   export_nokeepdimsz!SplitToSequence.export_nokeepdims=   s    y}}$$V,,33BJ??{$$HG % 
 
 GFFFtz!}1E1EFFFG6$4		
 	
 	
 	
 	
 	
r-   N)r   r	   )__name__
__module____qualname__staticmethodr,   r0   r;   r4   r-   r+   r   r      sr        
 
 
 \
. 
 
 
 \
, 
 
 
 \
 
 
r-   r   )

__future__r   numpyr   r%   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r4   r-   r+   <module>rD      s    # " " " " "      , , , , , , . . . . . .C
 C
 C
 C
 C
d C
 C
 C
 C
 C
r-   