PytorchBackend.topk_indices#

PytorchBackend.topk_indices(arr, k)[source]#

Determinces the indices of largest elements.

Parameters:
arrBackendArray

Input data.

kint

Number of maxima to determine.

Returns:
BackendArray

Indices of k largest elements in arr.