arrayfire/src/api/cpp/topk.cpp at master · arrayfire/arrayfire · GitHub

Repository navigation

27 lines (23 loc) · 791 Bytes

/*******************************************************

* This file is distributed under 3-clause BSD license.

* The complete license agreement can be obtained at:

* http://arrayfire.com/licenses/BSD-3-Clause

********************************************************/

#include <af/array.h>

#include <af/dim4.hpp>

#include <af/statistics.h>

#include "common.hpp"

#include "error.hpp"

namespace af {

void topk(array &values, array &indices, const array &in, const int k,

const int dim, const topkFunction order) {

af_array af_vals = 0;

af_array af_idxs = 0;

AF_THROW(af_topk(&af_vals, &af_idxs, in.get(), k, dim, order));

values = array(af_vals);

indices = array(af_idxs);

}

} // namespace af