
    sg
                    6   d dl mZ d dlZd dlZddlmZmZ ddlmZm	Z	 d Z
e	e
ddd              Ze	e
ddd              Ze	e
ddd	              Ze	e
ddd
              Ze	e
ddd              Ze	e
ddd              Ze	e
ddd              Ze	e
ddd              Ze	e
ddd              Ze	e
ddd              Ze	e
ddd              Ze	e
ddd              Ze	e
ddd              Ze	e
ddd              Ze	dd       Ze	dd       Ze	ddd       Ze	ddd       Zy)     )annotationsN   )_dtypes_impl_util)	ArrayLike
normalizerc                B     t        j                          fd       }|S )z>NumPy fft casts inputs to 64 bit and *returns 64-bit results*.c                    | j                         rt        j                         j                  nt        j                         j                  }t        j                  | |      }  | g|i |S N)
is_complexr   default_dtypescomplex_dtypefloat_dtyper   cast_if_needed)tensorargskwdstarget_dtypefuncs       C/var/www/html/venv/lib/python3.12/site-packages/torch/_numpy/fft.pywrappedzupcast.<locals>.wrapped   se       " '')77,,.:: 	
 %%fl;F*T*T**    )	functoolswraps)r   r   s   ` r   upcastr      s'     __T+ + Nr   c                H    t         j                  j                  | |||      S N)dimnorm)torchfftanaxisr   s       r   r!   r!      s     99==A4d=33r   c                H    t         j                  j                  | |||      S r   )r    r!   ifftr"   s       r   r'   r'   #        99>>!QDt>44r   c                H    t         j                  j                  | |||      S r   )r    r!   rfftr"   s       r   r*   r*   )   r(   r   c                H    t         j                  j                  | |||      S r   )r    r!   irfftr"   s       r   r,   r,   /        99??1aT?55r   c                H    t         j                  j                  | |||      S r   )r    r!   fftnr#   saxesr   s       r   r/   r/   5   r(   r   c                H    t         j                  j                  | |||      S r   )r    r!   ifftnr0   s       r   r4   r4   ;   r-   r   c                H    t         j                  j                  | |||      S r   )r    r!   rfftnr0   s       r   r6   r6   A   r-   r   c                H    t         j                  j                  | |||      S r   )r    r!   irfftnr0   s       r   r8   r8   G   "     99Aqd66r   c                H    t         j                  j                  | |||      S r   )r    r!   fft2r0   s       r   r;   r;   M   r(   r   c                H    t         j                  j                  | |||      S r   )r    r!   ifft2r0   s       r   r=   r=   S   r-   r   c                H    t         j                  j                  | |||      S r   )r    r!   rfft2r0   s       r   r?   r?   Y   r-   r   c                H    t         j                  j                  | |||      S r   )r    r!   irfft2r0   s       r   rA   rA   _   r9   r   c                H    t         j                  j                  | |||      S r   )r    r!   hfftr"   s       r   rC   rC   e   r(   r   c                H    t         j                  j                  | |||      S r   )r    r!   ihfftr"   s       r   rE   rE   k   r-   r   c                B    t         j                  j                  | |      S r   )r    r!   fftfreqr$   ds     r   rG   rG   q   s    99Q""r   c                B    t         j                  j                  | |      S r   )r    r!   rfftfreqrH   s     r   rK   rK   v   s    99a##r   c                B    t         j                  j                  | |      S r   )r    r!   fftshiftxr2   s     r   rM   rM   {   s    99a&&r   c                B    t         j                  j                  | |      S r   )r    r!   	ifftshiftrN   s     r   rQ   rQ      s    99q$''r   )NN)r#   r   )NNN)N)rR   N)g      ?r   )rO   r   )
__future__r   r   r     r   r   _normalizationsr   r   r   r!   r'   r*   r,   r/   r4   r6   r8   r;   r=   r?   rA   rC   rE   rG   rK   rM   rQ    r   r   <module>rX      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   