convolve_fft_torch

convolve_fft_torch#

jolideco.utils.torch.convolve_fft_torch(image, kernel)[source]#

Convolve FFT for torch tensors

Parameters:
  • image (~torch.Tensor) – Image tensor

  • kernel (~torch.Tensor) – Kernel tensor

Returns:

result – Convolution result

Return type:

~torch.Tensor