File size: 198 Bytes
2595c46 | 1 2 3 4 5 6 7 8 9 10 | #pragma once
#include <torch/all.h>
namespace megablocks {
// Public interface function for computing histograms
torch::Tensor histogram(torch::Tensor x, int num_bins);
} // namespace megablocks |