kaffe 0.2.0

Pytorch, but make it rust :)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

//! Integration tests
use kaffe::Tensor;

#[test]
fn creation() {
    let a: Tensor<f32> = Tensor::eye(2);

    assert!(a.data == vec![1.0, 0.0, 0.0, 1.0]);
    assert!(a.size() == 4);
    assert!(a.shape == vec![2, 2]);
}

#[test]
fn matmul() {
    let a = Tensor::init(2f32, vec![2, 100]);
    let b = Tensor::init(3f32, vec![100, 2]);

    let c = a.matmul(&b).unwrap_or_else(|_| Tensor::default());

    assert!(c.shape == vec![3, 3]);
    assert!(c.size() == 9);
}