tomodrgn.fft.ht3_center_torch

ht3_center_torch(possibly_batched_vols: Tensor) → Tensor

Batch-vectorized 3-D DHT of even-box sized volume(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_vols – torch.tensor of volumes, shape ([N], D, D, D)

Returns:

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