kaffe 0.2.0

Pytorch, but make it rust :)
Documentation 
1
2
3
4
5
6
7
8
9
10

use kaffe::Tensor;

fn main() {
    let a: Tensor<f32> = Tensor::randomize(vec![2, 4]);
    let b: Tensor<f32> = Tensor::randomize(vec![4, 3]);

    if let Ok(tensor) = a.matmul(&b) {
        println!("{:?}", tensor.data);
    }
}