Sort
Arguments
- self
(Tensor) the input tensor.
- dim
(int, optional) the dimension to sort along
- descending
(bool, optional) controls the sorting order (ascending or descending)
- stable
(bool, optional) – makes the sorting routine stable, which guarantees that the order of equivalent elements is preserved.
sort(input, dim=-1, descending=FALSE) -> (Tensor, LongTensor)
Sorts the elements of the input tensor along a given dimension
in ascending order by value.
If dim is not given, the last dimension of the input is chosen.
If descending is TRUE then the elements are sorted in descending
order by value.
A namedtuple of (values, indices) is returned, where the values are the
sorted values and indices are the indices of the elements in the original
input tensor.
Examples
if (torch_is_installed()) {
x = torch_randn(c(3, 4))
out = torch_sort(x)
out
out = torch_sort(x, 1)
out
}
#> [[1]]
#> torch_tensor
#> -0.4781 -2.1664 0.3595 0.1047
#> 0.0030 0.5783 0.4910 0.8585
#> 0.8602 0.5895 0.6141 1.0658
#> [ CPUFloatType{3,4} ]
#>
#> [[2]]
#> torch_tensor
#> 1 1 3 1
#> 2 3 1 3
#> 3 2 2 2
#> [ CPULongType{3,4} ]
#>