
    [Th
                    z   S SK Jr  S SKrS SKrSSKJrJr  SSKJrJ	r	  S r
\	\
SSS jj5       5       r\	\
SSS jj5       5       r\	\
SSS	 jj5       5       r\	\
SSS
 jj5       5       r\	\
SSS jj5       5       r\	\
SSS jj5       5       r\	\
SSS jj5       5       r\	\
SSS jj5       5       r\	\
SSS jj5       5       r\	\
SSS jj5       5       r\	\
SSS jj5       5       r\	\
SSS jj5       5       r\	\
SSS jj5       5       r\	\
SSS jj5       5       r\	SS j5       r\	SS j5       r\	SSS jj5       r\	SSS jj5       rg)     )annotationsN   )_dtypes_impl_util)	ArrayLike
normalizerc                F   ^  [         R                  " T 5      U 4S j5       nU$ )z>NumPy fft casts inputs to 64 bit and *returns 64-bit results*.c                   > U R                  5       (       a  [        R                  " 5       R                  O[        R                  " 5       R                  n[
        R                  " X5      n T" U /UQ70 UD6$ N)
is_complexr   default_dtypescomplex_dtypefloat_dtyper   cast_if_needed)tensorargskwdstarget_dtypefuncs       H/var/www/auris/envauris/lib/python3.13/site-packages/torch/_numpy/fft.pywrappedupcast.<locals>.wrapped   sf       "" '')77,,.:: 	
 %%f;F*T*T**    )	functoolswraps)r   r   s   ` r   upcastr      s'     __T+ + Nr   c                >    [         R                  R                  XX#S9$ N)dimnorm)torchfftanaxisr    s       r   r"   r"      s     99==4=33r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   ifftr#   s       r   r(   r(   #        99>>!D>44r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   rfftr#   s       r   r+   r+   )   r)   r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   irfftr#   s       r   r-   r-   /        99??1T?55r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   fftnr$   saxesr    s       r   r0   r0   5   r)   r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   ifftnr1   s       r   r5   r5   ;   r.   r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   rfftnr1   s       r   r7   r7   A   r.   r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   irfftnr1   s       r   r9   r9   G        99Ad66r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   fft2r1   s       r   r<   r<   M   r)   r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   ifft2r1   s       r   r>   r>   S   r.   r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   rfft2r1   s       r   r@   r@   Y   r.   r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   irfft2r1   s       r   rB   rB   _   r:   r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   hfftr#   s       r   rD   rD   e   r)   r   c                >    [         R                  R                  XX#S9$ r   )r!   r"   ihfftr#   s       r   rF   rF   k   r.   r   c                @    [         R                  R                  X5      $ r   )r!   r"   fftfreqr%   ds     r   rH   rH   q   s    99Q""r   c                @    [         R                  R                  X5      $ r   )r!   r"   rfftfreqrI   s     r   rL   rL   v   s    99a##r   c                @    [         R                  R                  X5      $ r   )r!   r"   fftshiftxr3   s     r   rN   rN   {   s    99a&&r   c                @    [         R                  R                  X5      $ r   )r!   r"   	ifftshiftrO   s     r   rR   rR      s    99q''r   )NN)r$   r   )NNN)N)rS   N)g      ?r   )rP   r   )
__future__r   r   r!    r   r   _normalizationsr   r   r   r"   r(   r+   r-   r0   r5   r7   r9   r<   r>   r@   rB   rD   rF   rH   rL   rN   rR    r   r   <module>rY      s   #   ! 2  4  4 5  5 5  5 6  6 5  5 6  6 6  6 7  7 5  5 6  6 6  6 7  7 5  5 6  6 # # $ $ ' ' ( (r   