polyTEM.image.fft.tensor_fft

polyTEM.image.fft.tensor_fft(tensor: Tensor, shape, centered=False)

Computes fft of 2D tensor using pytorch GPU implementation TENSOR FFT IS NOT ZERO CENTERED BY DEFAULT! the comment out is legacy code from camila :param tensor: 2D torch.tensor :param size: int shape of fft output size

Returns:

torch.Tensor of real component of fft imag: torch.Tensor of imaginary component of fft

Return type:

real