Torch.Indef.Static.Tensor.TopK

Documentation

topk :: forall d' d n. (All Dimensions '[d, d'], KnownDim n) => Tensor d -> Integer -> Word -> TopKOrder -> Maybe KeepDim -> (Tensor d', IndexTensor '[n])

returns all k smallest elements in a tensor over a given dimension, including their indices, in unsorted order.