ruPRIM-host 0.1.27

ruPRIM CPU tensor primitives, reductions, indexing and SIMD kernels.
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
72
73
74
75
76
77
78
79
80
81
82
83
84

use ruda_core::tensor::{host::HostTensor, element::Scalar};
use num_traits::ToPrimitive;
use crate::unary;
#[cfg(not(feature = "std"))]
#[allow(unused_imports)]
use num_traits::Float;

pub fn float_neg(tensor: HostTensor) -> HostTensor {
    unary::unary_op(tensor, |x: f32| -x, |x: f64| -x)
}

pub fn float_clamp(tensor: HostTensor, min: Scalar, max: Scalar) -> HostTensor {
    let min32 = min.to_f32().unwrap();
    let max32 = max.to_f32().unwrap();
    let min64 = min.to_f64().unwrap();
    let max64 = max.to_f64().unwrap();
    unary::unary_op(
        tensor,
        move |x: f32| x.clamp(min32, max32),
        move |x: f64| x.clamp(min64, max64),
    )
}

pub fn float_clamp_min(tensor: HostTensor, min: Scalar) -> HostTensor {
    let min32 = min.to_f32().unwrap();
    let min64 = min.to_f64().unwrap();
    unary::unary_op(
        tensor,
        move |x: f32| x.max(min32),
        move |x: f64| x.max(min64),
    )
}

pub fn float_clamp_max(tensor: HostTensor, max: Scalar) -> HostTensor {
    let max32 = max.to_f32().unwrap();
    let max64 = max.to_f64().unwrap();
    unary::unary_op(
        tensor,
        move |x: f32| x.min(max32),
        move |x: f64| x.min(max64),
    )
}

pub fn float_sign(tensor: HostTensor) -> HostTensor {
    unary::unary_op(
        tensor,
        |x: f32| {
            if x.is_nan() {
                x
            } else if x > 0.0 {
                1.0
            } else if x < 0.0 {
                -1.0
            } else {
                0.0
            }
        },
        |x: f64| {
            if x.is_nan() {
                x
            } else if x > 0.0 {
                1.0
            } else if x < 0.0 {
                -1.0
            } else {
                0.0
            }
        },
    )
}

pub fn float_is_nan(tensor: HostTensor, out_dtype: ruda_core::tensor::BoolDType) -> HostTensor {
    unary::float_predicate(tensor, out_dtype, |x: f32| x.is_nan(), |x: f64| x.is_nan())
}

pub fn float_is_inf(tensor: HostTensor, out_dtype: ruda_core::tensor::BoolDType) -> HostTensor {
    unary::float_predicate(
        tensor,
        out_dtype,
        |x: f32| x.is_infinite(),
        |x: f64| x.is_infinite(),
    )
}