ariadnetor-tensor 0.0.4

Tensor library with support for dense and block-sparse tensors
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

//! Tests for replace_slice operations.

use ariadnetor_tensor::{DenseTensorData, MemoryOrder};

#[test]
fn test_replace_slice_column_major() {
    // CM 4x4 zeros, write 2x2 sub at (1, 1)
    // CM sub for [[1,2],[3,4]]: col0=[1,3], col1=[2,4] -> flat [1,3,2,4]
    let mut t =
        DenseTensorData::<f64>::from_raw_parts(vec![0.0; 16], vec![4, 4], MemoryOrder::ColumnMajor);
    let sub = DenseTensorData::<f64>::from_raw_parts(
        vec![1.0, 3.0, 2.0, 4.0],
        vec![2, 2],
        MemoryOrder::ColumnMajor,
    );
    t.replace_slice(&sub, &[1, 1]);
    // In CM 4x4: col j is at flat[j*4..j*4+4].
    // (1,1) -> flat[1*1 + 4*1] = flat[5]; but CM: flat[col*rows + row] = flat[1*4 + 1] = flat[5]
    assert_eq!(t.data()[5], 1.0); // (1,1)
    assert_eq!(t.data()[6], 3.0); // (2,1)
    assert_eq!(t.data()[9], 2.0); // (1,2)
    assert_eq!(t.data()[10], 4.0); // (2,2)
    assert_eq!(t.data()[0], 0.0); // (0,0) untouched
    assert_eq!(t.data()[15], 0.0); // (3,3) untouched
}

#[test]
fn test_replace_slice_cow() {
    // Verify CoW: cloned tensor should not be affected
    let mut a =
        DenseTensorData::<f64>::from_raw_parts(vec![0.0; 4], vec![2, 2], MemoryOrder::ColumnMajor);
    let b = a.clone();
    let sub =
        DenseTensorData::<f64>::from_raw_parts(vec![1.0], vec![1, 1], MemoryOrder::ColumnMajor);
    a.replace_slice(&sub, &[0, 0]);
    assert_eq!(a.data()[0], 1.0);
    assert_eq!(b.data()[0], 0.0); // b should be unaffected
}