Function one_bit_code

include/rabitqlib/quantization/rabitq_impl.hpp:39–54

template <typename T>
inline RowMajorArray<T> one_bit_code(
    const T* data, const T* centroid, size_t dim, int* binary_code
) {
    // map pointer to array
    ConstRowMajorArrayMap<T> data_arr(data, 1, dim);
    ConstRowMajorArrayMap<T> cent_arr(centroid, 1, dim);

    // residual vector
    RowMajorArray<T> residual_arr = data_arr - cent_arr;

    // unsigned representation
    RowMajorArrayMap<int> x_u(binary_code, 1, static_cast<long>(dim));
    x_u = (residual_arr > 0).template cast<int>();

    return residual_arr;
}

/**
 * @brief The one_bit_code_with_factor function maps a data vector to a binary code and
 * computes factors for distance estimation.