search

teamtomo / libtilt

Image processing for cryo-electron microscopy in PyTorch
Other
14 stars 5 forks source link

use vmap instead of list comprehension in real space projection code #56

Open alisterburt opened 8 months ago

alisterburt commented 8 months ago

https://github.com/teamtomo/libtilt/blob/958e58096746a00e437f8714eca11815af4a669e/src/libtilt/projection/project_real.py#L76

@rsanchezgarc pointed out we should be able to use torch.vmap here rather than the list comprehension