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 Constant {
    v: f32,
}

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

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

pub fn constant(v: f32) -> Box<dyn Initializer> {
    Box::new(Constant { v })
}