matten-ndarray 0.20.1

Production-ready candidate conversion bridge between matten::Tensor and ndarray::ArrayD<f64>.
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

//! Convert an `ndarray::ArrayD<f64>` into a `matten::Tensor`, including a
//! non-standard-layout (transposed) input.
//!
//! Run:
//! cargo run -p matten-ndarray --example from_arrayd

use matten_ndarray::from_arrayd;
use ndarray::{ArrayD, IxDyn};

fn main() {
    let arr = ArrayD::from_shape_vec(IxDyn(&[2, 3]), vec![1., 2., 3., 4., 5., 6.]).unwrap();

    let t = from_arrayd(arr.clone()).expect("contiguous converts");
    println!(
        "from contiguous: shape {:?} data {:?}",
        t.shape(),
        t.as_slice()
    );

    // Transposed input is non-standard layout; conversion preserves logical order.
    let tt = from_arrayd(arr.t().to_owned()).expect("transposed converts");
    println!(
        "from transposed: shape {:?} data {:?}",
        tt.shape(),
        tt.as_slice()
    );
    assert_eq!(tt.shape(), &[3, 2]);
    assert_eq!(tt.as_slice(), &[1.0, 4.0, 2.0, 5.0, 3.0, 6.0]);
    println!("ok");
}