polyTEM.image.fft.rotate_filter

polyTEM.image.fft.rotate_filter(single_filter, angles, device='cuda')

Generates 3D tensor of size (M, M, number_angles) with stack of rotated Fourier filters to be used for datacube extraction.

Parameters:

  • single_filter – Initial filter as numpy array of size (M, M)

  • angles – list or numpy array with rotation angles

  • device – CUDA device send final output tensor

Returns:

3D filters tensor with series of filters for azimuthal information extraction

Return type:

tensor