torch_argsort {torch} | R Documentation |
Argsort
Description
Argsort
Usage
torch_argsort(self, dim = -1L, descending = FALSE)
Arguments
self |
(Tensor) the input tensor. |
dim |
(int, optional) the dimension to sort along |
descending |
(bool, optional) controls the sorting order (ascending or descending) |
argsort(input, dim=-1, descending=False) -> LongTensor
Returns the indices that sort a tensor along a given dimension in ascending order by value.
This is the second value returned by torch_sort
. See its documentation
for the exact semantics of this method.
Examples
if (torch_is_installed()) {
a = torch_randn(c(4, 4))
a
torch_argsort(a, dim=1)
}
[Package torch version 0.13.0 Index]