polyTEM.image.fft.get_orientation_torch

polyTEM.image.fft.get_orientation_torch(fft, filters, device)

Gets Intensity tensor for different angles at any grid point (x,y). Uses broadcasting and torch operations to speed 2x process with respect to loop.

Written by Camila

Parameters:

• fft – GPU torch tensor of fourier transform

• filters – GPU torch tensor of gaussian filters to be applied (different angles)

• device – CUDA device

Returns:

intensity tensor at grid point x,y