
    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                   &    e Zd Zedd            ZdS )STFTreturnNc                     t          j        ddt           j                                      ddd          } t          j        d                              t           j                  }|dz	  dz   }t          j        d                              t           j                  }d}t          j        	                    dd	d
|dgdg          }| j
        d         |z
  |z  dz   }t          j        d||dgt           j                  }t          |          D ]h}||z  }	||z  |z   }
t           j                            | d|	|
df                   d|         }t          j        |j        |j        fd          |d|f<   it#          || ||g|gd           t          j        	                    dg ddg          }d}d}||t          j        dt           j        z  t          j        d|dt           j                  z  |z            z  z   }d| j
        d         |j
        d         z
  |z  z   }t          j        d||dgt           j                  }t          |          D ]k}||z  }	||z  |z   }
t           j                            | d|	|
df         |z            d|         }t          j        |j        |j        fd          |d|f<   lt#          || ||g|gd           d S )Nr      )dtype          r   signal
frame_stepframe_lengthoutput)inputsoutputs   )axis	test_stft)r   r   name)r   r   windowg      ?test_stft_with_window)nparangefloat32reshapearrayastypeint64onnxhelper	make_nodeshapeemptyrangefftstackrealimagr   cospi)r   lengthonesided_lengthstep	no_windownodenstftsr   istartstopcomplex_outa0a1r   s                  \/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/stft.pyexportzSTFT.export   s   1c444<<QQGG"$$RX..!Q;!+x{{!!"(++	{$$lI~FJ % 
 
 <?V+491foq9LLLv 	R 	RAHEt8f$D&**VAuTz1,<%=>>q?PQK8[%5{7G$HqQQQF1a4LLtVT62VH;WWWW{$$555J % 
 
 b26I	!VQbjAAAAFJ
 
 
 
 fl1oQ7D@@ 1foq9LLLv 	R 	RAHEt8f$D&**VAuTz1,<%=%FGG/!K 8[%5{7G$HqQQQF1a4LLD&)H(		
 	
 	
 	
 	
 	
    )r   N)__name__
__module____qualname__staticmethodr;    r<   r:   r   r      s2        4
 4
 4
 \4
 4
 4
r<   r   )numpyr   r"   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rA   r<   r:   <module>rE      st         , , , , , , . . . . . .6
 6
 6
 6
 6
4 6
 6
 6
 6
 6
r<   