IЦi!F SSKrSSKJrJrJrJrJrJrJrJ r SSK r SSK J r SSKJr SSKJr SSKJrJrJr SSKJrJr /SQr\ S\S4r1Skr\ R8R:R<rS \S \S \S \ S \4 Sjr!S\ RDS\ S\ RFS \ RD4Sjr$SCS\S\ S \4Sjjr%S \S\\S4S\\S4S \4Sjr&S\'S\S\\S\S \S \ S \4Sjr(S\'S\S\\S\S \S \ S\ S \4Sjr)S\'S\S\\S\S \S \ S \4S jr*\"\RV5\"5SDS\S\\S\S \S \4 S!jj55r,\"\RZ5\"5SDS\S\\S\S \S \4 S"jj55r.\"\R^5\"5SDS\S\\S\S \S \4 S#jj55r0\"\Rb5\"5SDS\S\\S\S \S \4 S$jj55r2\"\Rf5\"5SDS\S\\S\S \S \4 S%jj55r4\"\Rj5\"5SDS\S\\S\S \S \4 S&jj55r6"S'S(\5r7S\S)\\S\\S \74S*jr8S+\\S \4S,jr9S-\'S\S)\\S4S\\S4S \S \ S \4S.jr:\"\Rv5\"5SES\S/\\S\\S \S \4 S0jj55r<\"\Rz5\"5SES\S/\\S\\S \S \4 S1jj55r>\"\R~5\"5SES\S/\\S\\S \S \4 S2jj55r@\"\R5\"5SES\S/\\S\\S \S \4 S3jj55rB"S4S5\5rCS6\'S\S/\\S\\S \C4 S7jrD\"\R5\"5SES\S/\\S\\S \S \4 S8jj55rF\"\R5\"5SES\S/\\S\\S \S \4 S9jj55rH\"\R5\"5SFS\S/\\S\\S \S \4 S:jj55rJ\"\R5\"5SFS\S/\\S\\S \S \4 S;jj55rL\"\R5\"5SFS\S/\\S\\S \S \4 S<jj55rN\"\R5\"5SFS\S/\\S\\S \S \4 S=jj55rP\"\R5\"5SFS\S/\\S\\S \S \4 S>jj55rR\"\R5\"5SFS\S/\\S\\S \S \4 S?jj55rTS\\S \S \\4S@jrU\"\R5SGS\S\\S \4SAjj5rW\"\R5SGS\S\\S \4SBjj5rYg)HN)IterableListLiteral NamedTupleOptionalSequenceTupleUnion)register_decomposition)DimsType ShapeTypeTensorLikeType)_maybe_convert_to_dtype out_wrapper)fftfft2fftnhffthfft2hfftnrfftrfft2rfftnifftifft2ifftnihfftihfft2ihfftnirfftirfft2irfftnfftshift ifftshift)forwardbackwardortho>Nr'r%r&xnorm signal_numelr%returnc^[R"T[;U4Sj5 TS:XaUS[R"U5- -$U(+=(a TSL=(d TS:H=(d U=(a TS:HnU(aUSU- -$U$)z3Apply normalization to the un-normalized FFT resultc>ST3$)NzInvalid normalization mode: )r)sN/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/_refs/fft.py_apply_norm...s 1MdV/Tr'Nr&r%)torch_check _NORM_VALUESmathsqrt)r(r)r*r% normalizes ` r/ _apply_normr:*s LL%'TU wA ,//00E$$,"D$*2D%DI%&/1L !5A5r2dtyperequire_complexdevicec^TR(aT$TR(d[R"5m[R[R /nUR S;nU(aUR[R5 [R"TU;U4Sj5 U(a[R"T5mT$)z@Helper to promote a dtype to one supported by the FFT primitives)cudametac>ST3$)NzUnsupported dtype r.r;sr/r0#_promote_type_fft..Is 3EeW1Mr2) is_complexis_floating_pointr4get_default_dtypefloat32float64typeappendfloat16r5utilscorresponding_complex_dtype)r;r<r= allowed_typesmaybe_support_halfs` r/_promote_type_fftrP9s    " "'')]]EMM2M(88U]]+ LL-')MN11%8 Lr2tc\URn[X!UR5n[X5$)zEHelper to promote a tensor to a dtype supported by the FFT primitives)r;rPr=r)rQr<cur_typenew_types r/_maybe_promote_tensor_fftrUQs(wwH AHHEH "1 //r2dims.sizesc[U5[U5:XdeSnURnS/[U5-S-n[[U55HfnX&S:XaM XAUX&:a&Sn[U5SX-- S- nX&XAU- XW'XAUX&:dMOURXSX&5nMh U(a[R "X5$U$)zv Fixes the shape of x such that x.size(dims[i]) == sizes[i], either by zero-padding, or by slicing x starting from 0. FrTr3)lenshaperangenarrowr4constant_pad_nd)r(rVrW must_copyx_sizes pad_amountipad_idxs r/_resize_fft_inputreZs t9E "" "IggGs7|#a'J 3t9  8r>  7 eh &I*oDG 3a7G"'(W!W-="=J  7 eh &!UX.A4=5  /C!Cr2 func_nameinputndimcl^[USS9n[R"URUSS94nUbUOSURUS- -m[ R "TS:U4Sj5 Ub[XTS-S-4S9nU(a[ R"U5n[R"XTS 9n[XtTUS 9$) zBCommon code for performing any complex to real FFT (irfft or hfft)Tr<F wrap_scalarrYr3c>STS3$NzInvalid number of data points (z ) specifiedr. last_dim_sizesr/r0_fft_c2r..1- Lr2)rVrWrirqr)r*r%) rUrLcanonicalize_dimndimr\r4r5reconjprimsfft_c2rr:) rfrgrhrir)r%rVoutputrqs @r/_fft_c2rr|ts &eT BE  " "5::s F HDAAS1AA1E,FM LLL  }!%=A;MPQ;Q:ST 5! ]]5- HF v}g VVr2onesidedc^^^ [R"TRR(+UU4Sj5 [ T5m[ R "TRUSS94nUbUOTRUm [R"T S:U 4Sj5 Ub [TXr45m[R"TXvS9n[XT U5nU(aU$[R"U5$)zBCommon code for performing any real to complex FFT (rfft or ihfft)c&>TSTR3$)Nz0 expects a floating point input tensor, but got rBrfrgsr/r0_fft_r2c..s9+Mekk][r2Frlr3c>STS3$ror.dim_sizesr/r0r!@ +Vr2rir})r4r5r;rDrUrLrvrwr\reryfft_r2cr:rx) rfrgrhrir)r%r}rVretrs `` @r/_fft_r2crs LL KK " ""[ &e ,E  " "5::s F HDMqu{{3'7H LLA V }!%t4 --4 ;C c7 3C3.uzz#.r2cr^^^[R"TRRUU4Sj5 [R "TR USS94nUbUOTRUm[R"TS:U4Sj5 Ub [TXb45m[R"TXeS9n[XtTU5$)zCCommon code for performing any complex to complex FFT (fft or ifft)c&>TSTR3$Nz) expects a complex input tensor, but got rBrsr/r0_fft_c2c..s9+Fu{{mTr2Frlr3c>STS3$ror.rsr/r0rrr2rir%) r4r5r;rDrLrvrwr\reryfft_c2cr:) rfrgrhrir)r%rVrrs `` @r/_fft_c2crs LL T  " "5::s F HDMqu{{3'7H LLA V }!%t4 --4 9C s(G 44r2c jURR(a [SXX#SS9$[SXX#SSS9$)NrTr%Fr%r}r;rDrrrgrhrir)s r/rrs5 {{uDAAuD5QQr2c jURR(a [SXX#SS9$[SXX#SSS9$)NrFrrrrs r/rrs5 {{#UCC#UUSSr2c [SXX#SSS9$)NrTrrrs r/rrs FEc MMr2c [SXX#SS9$)Nr Frr|rs r/r r s GUs% @@r2c [SXX#SS9$)NrTrrrs r/rrs FEc >>r2c [SXX#SSS9$)NrFTrrrs r/rrs GUs%$ OOr2c>\rSrSr%\\S4\S'\\S4\S'Srg) _ShapeAndDimsi.r\rVr.N__name__ __module__ __qualname____firstlineno__r int__annotations____static_attributes__r.r2r/rrs c? S/r2rr\cH^^^^URmURmUbd[U[5(dU4n[R "TUSS9n[ R"[[U55[U5:HS5 Ub[U[5(dU4n[ R"USL=(d [U5[U5:HS5 [U5m[ R"TT:*UU4Sj5 Uc[[TT- T55n[U4Sj[UW555nO8Uc [[T55n[T5nO[U4SjW55nUH!m[ R"TS :U4S j5 M# [XCS 9$) zTConvert the shape and dim arguments into a canonical form where neither are optionalNFrlcg)NzFFT dims must be uniquer.r.r2r/r06_canonicalize_fft_shape_and_dim_args..!s9Rr2cg)Nz=When given, dim and shape arguments must have the same lengthr.r.r2r/r0r+sSr2c>STSTS3$)NzGot shape with z" values but input tensor only has z dimensions.r.) input_dimtransform_ndimsr/r0r1son%56!{,0r2c3B># UHupUS:waUOTUv M g7f)rZNr.).0sd input_sizess r/ 7_canonicalize_fft_shape_and_dim_args..:s& ;O!bAk!n ,;Oc3.># UH nTUv M g7fNr.)rrrs r/rrCs;(Q+a.(src>STS3$ror.)rhsr/r0rFs&EaS $Tr2)r\rV)rwr\ isinstancerrLcanonicalize_dimsr4r5r[settupler]zipr) rgr\riret_dims ret_shaperrrhrs @@@@r/$_canonicalize_fft_shape_and_dim_argsrsm I++K #x((&C**9cuM  H #h- /1R  %**HE  4K 13s8s5z1 S U i ' 0  ;U9~#=yIJH ;>uh;O   y)*+& ;(;;  QUTU y 88r2xsc$SnUHnX-nM U$)zCompute product of a listr3r.)rprodr(s r/_prodrKs D   Kr2 function_namec^^[R"TRRUU4Sj5 [ TX25n[ R "XcUS9n[Xt[U5US9$)zECommon code for n-dimensional complex to complex FFTs (fftn or ifftn)c&>TSTR3$rrB)rrgsr/r0_fftn_c2c..^s=/";;-!r2rru) r4r5r;rDreryrr:r)rrgr\rir)r%r(r{s`` r/ _fftn_c2crSsT LL  ! %,A ]]1w 7F vuU|W UUr2rc J[XU5upB[USS9n[SXTX#SS9$)NTrkrrrrUrrgrrir)r\r(s r/rrfs/8#FLU!%>A VQs$ ??r2c J[XU5upB[USS9n[SXTX#SS9$)NTrkrFrrrs r/rrss/8#FLU!%>A Wa5 AAr2c^[R"TRR(+U4Sj5 [ TX5upB[ TSS9m[ TX$5m[R"TUSS9n[XS[U5SS9$)Nc">STR3$)Nz2rfftn expects a real-valued input tensor, but got rBrgsr/r0rfftn..sDU[[MRr2FrkTrru) r4r5r;rDrrUreryrr:r)rgrrir)r\outs` r/rrsq LL KK " ""R6eQDJE %eU CE eS 0E --3 6C sE%L$ OOr2c^[R"TRR(+U4Sj5 [ TX5upB[R"[ U5S:S5 [ TSS9m[TX$5m[R"TUSSSS9n[ U5S :Xa$[XSUSSS 9n[R"U5$[R"U5n[R"XRSSSS 9n[XS[U5SS 9$) Nc">STR3$)Nz3ihfftn expects a real-valued input tensor, but got rBrsr/r0ihfftn..sEekk]Sr2rcg)Nz'ihfftn must transform at least one axisr.r.r2r/r0rs)Rr2FrkrZTrr3rur)r4r5r;rDrr[rUreryrr:rx conj_physicalrr)rgrrir)r\tmps` r/rrs LL KK " ""S6eQDJE LLUa!RS %eU CE eS 0E --3rs8d ;C 3x1}#uQxOzz#   c "C --Sb5 9C sE%L% PPr2cH\rSrSr%\\S4\S'\\S4\S'\\S'Srg)_CanonicalizeC2rReturni.r\rirqr.Nrr.r2r/rrs$ c? sCxr2rfnamecT^^[XU5upC[R"[U5S:U4Sj5 Ub USS:XaSURUSS- -mOUSm[R"TS:U4Sj5 [ U5nTS-S-US'[ [U5UTS9$)zCanonicalize shape and dim arguments for n-dimensional c2r transforms, as well as calculating the last_dim_size which is shape[dim[-1]] for the outputrc>TS3$)Nz! must transform at least one axisr.)rsr/r0:_canonicalize_fft_c2r_shape_and_dim_args..s E72S)Tr2rZrYr3c>STS3$ror.rpsr/r0rrsr2)r\rirq)rr4r5r[r\listrr)rrgrrir\ shape_listrqs` @r/(_canonicalize_fft_c2r_shape_and_dim_argsrs8#FLU LLUa!TUyAbERKU[[R1A56 b  LLL eJ"a'!+JrN !JS  r2c^[SXU5upBn[USS9n[XU5n[R"XUS9m[ TU[ U4SjU55SS9$)Nr"Trkrtc3B># UHnTRUv M g7fr)r\)rrrs r/rirfftn..s'Bc ! crFr)rrUreryrzr:r)rgrrir)r\rqrs @r/r"r"sb!I%C!E  &eT BE e% 0E --m DC sD%'Bc'B"BE RRr2c @[SXU5upBn[USS9n[XU5n[U5S:a[R "XSSSS9OUn[ Xc[USS5SS9n[R"U5n[R"XbSSUS9n[ XsUSS9$) NrTrkr3rZrrrt) rrUrer[ryrr:rrrz)rgrrir)r\rqrrs r/rrs!I3!E  &eT BE e% 0E>A#hl%--s8T :PUC cuSbz!2D AC   c "C --RS GC s- >>r2c>[RRXX#S9$N)rrir))r4rrrgrrir)s r/rrs 99>>%#> 99r2c>[RRXX#S9$r)r4rrrs r/rr 99??53? ::r2c>[RRXX#S9$r)r4rrrs r/rr rr2c>[RRXX#S9$r)r4rr"rs r/r!r! 99  EC  ;;r2c>[RRXX#S9$r)r4rrrs r/rr#rr2c>[RRXX#S9$r)r4rrrs r/rr.rr2cUc[[UR55$[U[5(dU/$[U5$)zIConvert Optional[DimsType] to a simple list, defaulting to all dimensions)rr]rwrr)rir(s r/_default_alldimsr9s8 {E!&&M"" X & &u Cyr2c[X5nUVs/sHo0RUS-PM nn[R"XU5$s snf)NrYrr\r4rollrgrirVrshifts r/r#r#CsA C 'D*. /$Q[[^q $E / ::eD )) 0sAc[X5nUVs/sHo0RUS-S-PM nn[R"XU5$s snf)Nr3rYrrs r/r$r$JsF C 'D04 51kk!nq Q &E 5 ::eD )) 6sA)F)NrZN)NNN)N)rZNr)Zr7typingrrrrrrr r r4 torch._prims_primsrytorch._prims_common _prims_commonrL torch._decompr r r rtorch._prims_common.wrappersrr__all__NormTyper6_opsopsatenrboolr:r;r=rPrUrestrr|rrfft_fftrfft_ifftrfft_rfftr fft_irfftr fft_hfftr fft_ihfftrrrrrfft_fftnr fft_ifftnr fft_rfftnr fft_ihfftnrrr fft_irfftnr" fft_hfftnrfft_fft2r fft_ifft2r fft_rfft2r fft_irfft2r! fft_hfft2r fft_ihfft2rr fft_fftshiftr# fft_ifftshiftr$r.r2r/rs/  XXX #0CCM 2 w=>> ?5  zz~~ 6 6% 658 6CG 6 6 ;;)-7<|| [[20500(,00DD"38_D5:38_DD4WW W}W W  W  WW6// /}/ /  /  ///:55 5}5 5  5  554 %  R  R} R  R  R  R& R &  T  T} T  T  T  T' T &  N N}N N  N  N'N'  A A}A A  A  A(A &  ? ?}? ?  ?  ?'?'  P P}P P  P  P(PJ 59 59"*9"559