Closed jank324 closed 1 year ago
Used torch.histogramdd instead of custom cheetah.utils.histogramdd.
torch.histogramdd
cheetah.utils.histogramdd
The current implementation was chosen because PyTorch didn't have a histogramdd function at the time. Now that one exists, using it would probably be faster and certainly offer code maintenance advantages.
histogramdd
🚀 Feature Request
Used
torch.histogramddinstead of customcheetah.utils.histogramdd.Motivation
The current implementation was chosen because PyTorch didn't have a
histogramddfunction at the time. Now that one exists, using it would probably be faster and certainly offer code maintenance advantages.