
    Ng.                     J    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 )    N)Base)expectc                   V    e Zd Zedd            Zedd            Zedd            ZdS )SplitToSequencereturnNc                  :   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       g/var/www/html/ai-engine/env/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>H   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+   )r   N)__name__
__module____qualname__staticmethodr*   r.   r9   r2   r+   r)   r   r      sr        
 
 
 \
. 
 
 
 \
, 
 
 
 \
 
 
r+   r   )numpyr   r#   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r2   r+   r)   <module>rA      s~   
      , , , , , , . . . . . .C
 C
 C
 C
 C
d C
 C
 C
 C
 C
r+   