Python fft 예제들

예제 #1

파일: TheanoUtil.py 프로젝트: tazdriver/returnn

def circular_convolution(a, b):
    from Device import is_using_gpu
    has_gpuarray = is_using_gpu()
    try:
        import pygpu
    except Exception:
        has_gpuarray = False
    if has_gpuarray:
        from theano.gpuarray.fft import curfft as fft
        from theano.gpuarray.fft import cuirfft as ifft
    else:
        from theano.tensor.fft import rfft as fft
        from theano.tensor.fft import irfft as ifft
    return ifft(fft(a) * fft(b))

예제 #2

파일: TheanoUtil.py 프로젝트: atuxhe/returnn

def circular_convolution(a, b):
  from Device import is_using_gpu
  has_gpuarray = is_using_gpu()
  try:
    import pygpu
  except Exception:
    has_gpuarray = False
  if has_gpuarray:
    from theano.gpuarray.fft import curfft as fft
    from theano.gpuarray.fft import cuirfft as ifft
  else:
    from theano.tensor.fft import rfft as fft
    from theano.tensor.fft import irfft as ifft
  return ifft(fft(a) * fft(b))