tensorflow分桶API，有好几个接口，其中带boundaries的接口C++实现如下：
template <typename T>
struct BucketizeFunctor<CPUDevice, T> {
// PRECONDITION: boundaries_vector must be sorted.
static Status Compute(OpKernelContext* context,
const typename TTypes<T, 1>::ConstTensor& input,
const std::vector<float>& boundaries_vector,
typename TTypes<int32, 1>::Tensor& output) {
const int N = input.size();
for (int i = 0; i < N; i++) {
auto first_bigger_it = std::upper_bound(
boundaries_vector.begin(), boundaries_vector.end(), input(i));
output(i) = first_bigger_it - boundaries_vector.begin();
}
return Status::OK();
}
};
- 输入：input tensor和boundaries_vector
- 输出：output tensor
使用stl的upper_bound算法查找第一个大于输入值的bound，然后返回这个bound的偏移索引。
所以这里，用户需要指定一个boundary，并且boundary是不变的。
