auto-diff-ann 0.5.9

A neural network library in Rust.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12

use ::rand::prelude::StdRng;

use auto_diff::{Var, AutoDiffError};
use tensor_rs::tensor::Tensor;

pub fn normal(data: &Tensor, mean: Option<Var>, std: Option<Var>, rng: &mut StdRng) -> Result<(), AutoDiffError>{
    let size = data.size();
    let mean = if let Some(v) = mean {f64::try_from(v)?} else {0.};
    let std = if let Some(v) = std {f64::try_from(v)?} else {1.};
    data.swap(&Var::normal(rng, &size, mean, std).val());
    Ok(())
}