fn_der 0.1.3

A trait for functions that have a derivative
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

#![feature(unboxed_closures)]
#![feature(const_trait_impl)]
#![feature(const_mut_refs)]
#![feature(associated_type_defaults)]

use fn_grad::*;

#[const_trait]
pub trait FnDerOnce<Arg>: FnOnce<(Arg,)> + ~const FnGradOnce<(Arg,)>
{
    fn derivative_once(self, arg: Arg) -> Self::GradientOutput
    where
        Self: Sized
    {
        self.gradient_once((arg,))
    }
}

#[const_trait]
pub trait FnDerMut<Arg>: FnDerOnce<Arg> + FnMut<(Arg,)> + ~const FnGradMut<(Arg,)>
{
    fn derivative_mut(&mut self, arg: Arg) -> Self::GradientOutput
    {
        self.gradient_mut((arg,))
    }
}

#[const_trait]
pub trait FnDer<Arg>: FnDerMut<Arg> + Fn<(Arg,)> + ~const FnGrad<(Arg,)>
{
    fn derivative(&self, arg: Arg) -> Self::GradientOutput
    {
        self.gradient((arg,))
    }
}