kaffe 0.2.0

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

use kaffe::nn::activation::ReLU;
use kaffe::Tensor;

#[test]
fn relu() {
    let mat = Tensor::new(vec![-2.0, -4.0, 6.0], vec![3, 1]).unwrap();
    assert_eq!(ReLU(&mat).data, vec![0.0, 0.0, 6.0]);
}