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 |