
    h                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   (    \ rS rSr\SS j5       rSrg)STFT   Nc                     [         R                  " SS[         R                  S9R                  SSS5      n [         R                  " S5      R                  [         R                  5      nUS-	  S-   n[         R                  " S5      R                  [         R                  5      nSn[        R                  R                  SS	S
US/S/S9nU R                  S   U-
  U-  S-   n[         R                  " SXbS/[         R                  S9n[        U5       Hg  nX-  n	X-  U-   n
[         R                  R                  U SX2S4   5      SU n[         R                  " UR                  UR                   4SS9USU4'   Mi     [#        XPX1/U/SS9  [        R                  R                  S/ SQS/S9nSnSnX[         R$                  " S[         R&                  -  [         R                  " SUS[         R                  S9-  U-  5      -  -   nSU R                  S   UR                  S   -
  U-  -   n[         R                  " SXbS/[         R                  S9n[        U5       Hj  nX-  n	X-  U-   n
[         R                  R                  U SX2S4   U-  5      SU n[         R                  " UR                  UR                   4SS9USU4'   Ml     [#        UXU/U/SS9  g )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                  Z/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/stft.pyexportSTFT.export   s   1c4<<QQG"$$RXX.!Q;!+xx{!!"((+	{{$$lI~FJ % 
 <<?V+491fq9LvAHE8f$D&&**VAuz1,<%=>qQK88[%5%5{7G7G$HqQF1a4L	  	tT2VH;W{{$$5J % 
 266I		!VQbjjAAFJ
 
 
 fll1oQ7D@@ 1fq9LvAHE8f$D&&**VAuz1,<%=%FG/K 88[%5%5{7G7G$HqQF1a4L  	&)H(		
     )returnN)__name__
__module____qualname____firstlineno__staticmethodr;   __static_attributes__r>   r=   r:   r   r      s    4
 4
r=   r   )numpyr   r"   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r>   r=   r:   <module>rI      s      , .6
4 6
r=   