o 8wiJ @s2ddlZddlmZmZiZddZdddZdS) N)rfftirfftcCszt|WStyYnwt|tr|dksJ|} |}dD]}||dkr3||}||dks)q!|dkr>|t|<|S|d7}q)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: rT))_NEXT_FAST_LENKeyError isinstanceint)size next_size remainingnrd/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/torch_audiomentations/utils/convolution.py next_fast_lens&    rfullc Cs|d}|d}|dkr||d}n"|dkr&t||t||d}n|dkr0t||}ntd|||d}t|}t||d}t||d} || } t| |d} ||d} | d | | |fS) 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: rrvalidsamezUnknown mode: {})rr.)r maxmin ValueErrorformatrrr) signalkernelmodemrtruncate padded_size fast_ftt_sizef_signalf_kernelf_resultresult start_idxrrrconvolve(s"        r')r)torchtorch_audiomentations.utils.fftrrrrr'rrrrs