aesara.tensor.topk_and_argtopk

aesara.tensor.topk_and_argtopk#

aesara.tensor.topk_and_argtopk(x, kth, axis=-1, sorted=True, idx_dtype='int64')[source]#

Returns the results of both topk() and argtopk() in one Op.

See the respective documentation for details.

Returns:

tuple

Return type:

(values, indices)