redstone-ml 0.0.0

High-performance Machine Learning, Auto-Differentiation and Tensor Algebra crate for Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

use crate::gradient_function::{GradientFuncTrait, GradientFunction};
use crate::{NdArray, TensorDataType};
use std::cell::RefCell;
use std::rc::Rc;


/// The default backwards node for non-leaf NdArrays.
pub(crate) struct NoneBackwards {}

impl<T: TensorDataType> GradientFuncTrait<T> for NoneBackwards {
    /// Backwards method for ndarray with `requires_grad = false`, does nothing.
    fn backward(&mut self, _: &NdArray<T>) {}

    fn is_none(&self) -> bool {
        true
    }
}

impl NoneBackwards {
    pub(crate) fn new<T: TensorDataType>() -> GradientFunction<T> {
        Rc::new(RefCell::new(Self {}))
    }
}