fn_grad 0.1.3

A trait for functions that have a gradient
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

#![no_std]

#![feature(unboxed_closures)]
#![feature(tuple_trait)]
#![feature(associated_type_bounds)]
#![feature(const_trait_impl)]

use core::marker::Tuple;

#[const_trait]
pub trait FnGradOnce<Args>: FnOnce<Args>
where
    Args: Tuple
{
    type GradientOutput;

    fn gradient_once(self, args: Args) -> Self::GradientOutput;
}

#[const_trait]
pub trait FnGradMut<Args>: FnGradOnce<Args> + FnMut<Args>
where
    Args: Tuple
{
    fn gradient_mut(&mut self, args: Args) -> Self::GradientOutput;
}

#[const_trait]
pub trait FnGrad<Args>: FnGradMut<Args> + Fn<Args>
where
    Args: Tuple
{
    fn gradient(&self, args: Args) -> Self::GradientOutput;
}