acme-tensor 0.3.1

A comprehensive tensor library for Rust with support for automatic-differentiation.
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

/*
    Appellation: tri <mod>
    Contrib: FL03 <jo3mccain@icloud.com>
*/
pub use self::utils::*;

pub(crate) mod utils {
    use crate::linalg::UPLO;
    use crate::tensor::TensorBase;
    use num::Zero;

    pub fn triangular<T>(a: &TensorBase<T>, uplo: UPLO) -> TensorBase<T>
    where
        T: Clone + Zero,
    {
        debug_assert!(a.is_square(), "Tensor must be square.");
        match uplo {
            UPLO::Upper => triu(a),
            UPLO::Lower => tril(a),
        }
    }

    /// Returns the lower triangular portion of a matrix.
    pub fn tril<T>(a: &TensorBase<T>) -> TensorBase<T>
    where
        T: Clone + Zero,
    {
        debug_assert!(a.is_square(), "Matrix must be square.");
        let mut out = a.clone();
        for i in 0..a.shape()[0] {
            for j in i + 1..a.shape()[1] {
                out[[i, j]] = T::zero();
            }
        }
        out
    }
    /// Returns the upper triangular portion of a matrix.
    pub fn triu<T>(a: &TensorBase<T>) -> TensorBase<T>
    where
        T: Clone + Zero,
    {
        let mut out = a.clone();
        for i in 0..a.shape()[0] {
            for j in 0..i {
                out[[i, j]] = T::zero();
            }
        }
        out
    }
}