tomodrgn.fft.ht2_center_torch

ht2_center_torch(possibly_batched_imgs: Tensor) → Tensor

Batch-vectorized 2-D DHT of an even-box sized image(s) as torch tensor Uses relationship DHT(x) == DFT(x).real - DFT(x).imag == (DFT(x) * (1 + i)).real Note: torch maintains default input dtype torch.float32 to output dtype torch.float32

Parameters:

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

Returns:

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