gradients 0.3.4

An OpenCL, CUDA and CPU based Deep Learning Library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

use custos::CPU;
use gradients::{Conv2D, Matrix};

#[test]
fn test_conv() {
    let device = CPU::new();

    let inputs = Matrix::from((&device, 28, 28, [1.1; 28 * 28]));

    let _conv = Conv2D::<f32>::new(&device, (28, 28), (3, 3), 5);

    let mut conv = Conv2D::<f32>::new(&device, (28, 28), (3, 3), 5);
    let out = conv.forward(&inputs);
    conv.forward(&out);
}