Function nearestNeighbour

src/api/cpp/nearest_neighbour.cpp:16–25

namespace af {

void nearestNeighbour(array& idx, array& dist, const array& query,
                      const array& train, const dim_t dist_dim,
                      const unsigned n_dist, const af_match_type dist_type) {
    af_array temp_idx  = 0;
    af_array temp_dist = 0;
    AF_THROW(af_nearest_neighbour(&temp_idx, &temp_dist, query.get(),
                                  train.get(), dist_dim, n_dist, dist_type));
    idx  = array(temp_idx);
    dist = array(temp_dist);
}

}  // namespace af