Skip to main content

sort

axonml_tensor::ops

Function sort 

Source 
pub fn sort<T: Numeric>(
    x: &Tensor<T>,
    dim: i64,
    descending: bool,
) -> Result<SortResult<T>>
Expand description

Sorts the elements of the tensor along a dimension.

§Arguments

  • x - Input tensor
  • dim - Dimension to sort along (default: -1)
  • descending - If true, sort in descending order

§Returns

SortResult containing sorted values and indices