torch_ecg.utils.np_topk

torch_ecg.utils.np_topk(arr: ndarray, k: int, dim: int = -1, largest: bool = True, sorted: bool = True) → Tuple[ndarray, ndarray]

Find the k largest elements of an array along a specified axis.

Parameters:

• arr (array_like) – Input array.

• k (int) – Number of elements to retrieve.

• dim (int, default -1) – Axis along which to operate. Default is -1 (the last axis).

• largest (bool, default True) – If True, find the largest elements, else find the smallest elements.

• sorted (bool, default True) – If True, the result is sorted. If False, the result is not sorted.

Returns:

• values (numpy.ndarray) – The k largest values along each axis.

• indices (numpy.ndarray) – The indices of the k largest values along each axis.

• .. note:: – This function has the same functionality as torch.topk(), but is implemented using only numpy.