ClikaRT::ops::topk

function

topk()

std::array<Tensor 2> topk(
    Tensor x,
    int64_t k,
    int64_t dim = -1,
    bool largest = true,
    bool sorted = true
)

Returns: a 2-element array {values, indices}.

Declared in ClikaRT/compute/ops.h, line 4620