scirs2-autograd 0.3.2

Automatic differentiation module for SciRS2 (scirs2-autograd)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

use crate::op;

use crate::Float;

pub(crate) struct ControlDependency;

impl<F: Float> op::Op<F> for ControlDependency {
    // Reuse the 1st input
    fn compute(&self, ctx: &mut op::ComputeContext<F>) -> Result<(), op::OpError> {
        let ret = ctx.input(0);
        ctx.append_output(ret.to_owned());
        Ok(())
    }

    fn grad(&self, ctx: &mut op::GradientContext<F>) {
        for _ in 0..ctx.num_inputs() {
            ctx.append_input_grad(0, None);
        }
    }
}