gtensor 1.0.0

Reverse-mode autodifferentiation of computational graphs with tensors and more for machine learning.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20


use super::*;

struct Uniform {
    value: f32,
}

impl Initializer for Uniform {
    fn init_vec(&self, shape: Shape) -> Vec<f32> {
        vec![self.value; shape.len()]
    }

    fn _clone(&self) -> Box<dyn Initializer> {
        Box::new(Uniform { value: self.value })
    }
}

pub fn uniform(value: f32) -> Box<dyn Initializer> {
    Box::new(Uniform {value})
}