rstsr 0.7.1

An n-Dimension Rust Tensor Toolkit
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

#![doc = include_str!("../notes_to_api_doc.md")]
#![doc = include_str!("../readme.md")]
pub mod prelude;

#[allow(unused_imports)]
use crate as rstsr;

pub mod api_specification {
    #![doc = include_str!("docs/api_specification.md")]
    #![allow(unused_imports)]

    use crate::prelude::*;
    use rt::*;
    extern crate alloc;
}

pub mod array_api_standard {
    #![doc = include_str!("docs/array_api_standard.md")]
    #![doc = include_str!("docs/array_api_standard_extension.md")]
    #![allow(unused_imports)]

    use crate::prelude::*;
    use core::ops::*;
    use num::complex::ComplexFloat;
    use num::{pow::Pow, Complex, Float, Integer, Num, Signed};
    use rstsr_core::prelude_dev::Indexer;
    use rstsr_dtype_traits::*;
    use rt::*;
}

#[test]
fn test() {
    use crate::prelude::*;

    // 3x2 matrix with c-contiguous memory layout
    let a = rt::asarray((vec![6., 2., 7., 4., 8., 5.], [3, 2].c()));

    // 2x4x3 matrix by arange and reshaping
    let b = rt::arange(24.);
    let b = b.reshape((-1, 4, 3));
    // in one line, you can also
    // let b = rt::arange(24.).into_shape((-1, 4, 3));

    // broadcasted matrix multiplication
    let c = b % a;

    // print layout of the result
    println!("{:?}", c.layout());
    // output:
    // 3-Dim (dyn), contiguous: Cc
    // shape: [2, 4, 2], stride: [8, 2, 1], offset: 0

    // print the result
    println!("{c:6.1}");
    // output:
    // [[[   23.0   14.0]
    //   [   86.0   47.0]
    //   [  149.0   80.0]
    //   [  212.0  113.0]]
    //
    //  [[  275.0  146.0]
    //   [  338.0  179.0]
    //   [  401.0  212.0]
    //   [  464.0  245.0]]]
}