tomodrgn.fft.fft2_center_torch

fft2_center_torch(possibly_batched_imgs: Tensor) → Tensor

Batch-vectorized 2-D DFT of an even-box sized image(s) as torch tensor Note: torch maintains default input dtype torch.float32 to output dtype torch.float32

Parameters:

possibly_batched_imgs – torch.tensor of images DFT, shape ([N], D, D)

Returns:

torch.tensor of images, shape ([N], D, D)