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

use crate::gradient_function::GradientFunction;
use crate::{FloatDataType, Tensor};

pub(crate) struct IdentityBackwards {}

impl IdentityBackwards {
    pub(crate) fn new<T: FloatDataType>(tensor: &Tensor<T>) -> GradientFunction<T> {
        tensor.grad_fn()
    }
}