iJ .ddlZddlmZmZiZdZddZy)N)rfftirfftc t|S#t$rYnwxYwt|tr|dkDsJ|} |}dD]}||zdk(s ||z}||zdk(r|dk(r |t|<|S|dz }6)a7 Returns the next largest number ``n >= size`` whose prime factors are all 2, 3, or 5. These sizes are efficient for fast fourier transforms. Equivalent to :func:`scipy.fftpack.next_fast_len`. Note: This function was originally copied from the https://github.com/pyro-ppl/pyro repository, where the license was Apache 2.0. Any modifications to the original code can be found at https://github.com/asteroid-team/torch-audiomentations/commits :param int size: A positive number. :returns: A possibly larger number. :rtype int: r))_NEXT_FAST_LENKeyError isinstanceint)size next_size remainingns |/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torch_audiomentations/utils/convolution.py next_fast_lenrs d##     dC TAX- -I  Aa-1$a a-1$ >#,N4  Q  s  c|jd}|jd}|dk(r ||zdz }nN|dk(rt||t||z dz}n,|dk(r t||}ntdj |||zdz }t |}t ||}t ||} || z} t| |} ||z dz} | d | | |zfS) a Computes the 1-d convolution of signal by kernel using FFTs. The two arguments should have the same rightmost dim, but may otherwise be arbitrarily broadcastable. Note: This function was originally copied from the https://github.com/pyro-ppl/pyro repository, where the license was Apache 2.0. Any modifications to the original code can be found at https://github.com/asteroid-team/torch-audiomentations/commits :param torch.Tensor signal: A signal to convolve. :param torch.Tensor kernel: A convolution kernel. :param str mode: One of: 'full', 'valid', 'same'. :return: A tensor with broadcasted shape. Letting ``m = signal.size(-1)`` and ``n = kernel.size(-1)``, the rightmost size of the result will be: ``m + n - 1`` if mode is 'full'; ``max(m, n) - min(m, n) + 1`` if mode is 'valid'; or ``max(m, n)`` if mode is 'same'. :rtype torch.Tensor: fullr validsamezUnknown mode: {})rr.)rmaxmin ValueErrorformatrrr) signalkernelmodemrtruncate padded_size fast_ftt_sizef_signalf_kernelf_resultresult start_idxs rconvolver)(s(  BA BA v~q519 q!9s1ay(1, q!9+224899a%!)K!+.MFm,HFm,H("H 8} -Fx'A-I #y9x#777 88)r)torchtorch_audiomentations.utils.fftrrr rr)r*rr.s 7@)9r*