numr 0.5.2

High-performance numerical computing with multi-backend GPU acceleration (CPU/CUDA/WebGPU)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

//! CPU implementation of sorting operations.

use crate::error::Result;
use crate::ops::SortingOps;
use crate::runtime::cpu::{CpuClient, CpuRuntime};
use crate::tensor::Tensor;

/// SortingOps implementation for CPU runtime.
impl SortingOps<CpuRuntime> for CpuClient {
    fn sort(
        &self,
        a: &Tensor<CpuRuntime>,
        dim: isize,
        descending: bool,
    ) -> Result<Tensor<CpuRuntime>> {
        crate::runtime::cpu::sort::sort_impl(self, a, dim, descending)
    }

    fn sort_with_indices(
        &self,
        a: &Tensor<CpuRuntime>,
        dim: isize,
        descending: bool,
    ) -> Result<(Tensor<CpuRuntime>, Tensor<CpuRuntime>)> {
        crate::runtime::cpu::sort::sort_with_indices_impl(self, a, dim, descending)
    }

    fn argsort(
        &self,
        a: &Tensor<CpuRuntime>,
        dim: isize,
        descending: bool,
    ) -> Result<Tensor<CpuRuntime>> {
        crate::runtime::cpu::sort::argsort_impl(self, a, dim, descending)
    }

    fn topk(
        &self,
        a: &Tensor<CpuRuntime>,
        k: usize,
        dim: isize,
        largest: bool,
        sorted: bool,
    ) -> Result<(Tensor<CpuRuntime>, Tensor<CpuRuntime>)> {
        crate::runtime::cpu::sort::topk_impl(self, a, k, dim, largest, sorted)
    }

    fn unique(&self, a: &Tensor<CpuRuntime>, sorted: bool) -> Result<Tensor<CpuRuntime>> {
        crate::runtime::cpu::sort::unique_impl(self, a, sorted)
    }

    fn unique_with_counts(
        &self,
        a: &Tensor<CpuRuntime>,
    ) -> Result<(Tensor<CpuRuntime>, Tensor<CpuRuntime>, Tensor<CpuRuntime>)> {
        crate::runtime::cpu::sort::unique_with_counts_impl(self, a)
    }

    fn nonzero(&self, a: &Tensor<CpuRuntime>) -> Result<Tensor<CpuRuntime>> {
        crate::runtime::cpu::sort::nonzero_impl(self, a)
    }

    fn searchsorted(
        &self,
        sorted_sequence: &Tensor<CpuRuntime>,
        values: &Tensor<CpuRuntime>,
        right: bool,
    ) -> Result<Tensor<CpuRuntime>> {
        crate::runtime::cpu::sort::searchsorted_impl(self, sorted_sequence, values, right)
    }
}