flashlight_tensor 0.4.5

gpu/cpu tensor library focused around matrix and neural network operations
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

pub mod relu;
pub mod sigmoid;

pub fn forward_shape(weight_shapes: &[u32], input_shapes:&[u32]) -> Vec<u32>{
    vec!{weight_shapes[0], input_shapes[1]}
}
pub fn backward_weights_shape(weight_shapes: &[u32]) -> Vec<u32>{
    weight_shapes.to_vec()
}
pub fn backward_bias_shape(bias_shapes: &[u32]) -> Vec<u32>{
    bias_shapes.to_vec()
}
pub fn backward_grad_shape(cache_shape: &[u32]) -> Vec<u32>{
    cache_shape.to_vec()
}