Trait ktensor::Graph

pub trait Graph<T> where
    T: Copy, {
    fn get_id(&self) -> String;
    fn get_dim(&self) -> Vec2;
    fn run(&self, state: &Context<T>, variable: &Context<T>) -> Tensor<T>;
    fn forward_pass(
        &self, 
        state: &Context<T>, 
        variable: &Context<T>, 
        history: &mut Context<T>
    ) -> Tensor<T>;
    fn backward_pass(
        &self, 
        state: &mut Context<T>, 
        variable: &Context<T>, 
        history: &Context<T>, 
        gradient: &Tensor<T>, 
        learning_rate: T
    );

    fn train(
        &self, 
        state: &Context<T>, 
        variable: &Context<T>, 
        history: &mut Context<T>
    ) -> Tensor<T> { ... }
}

Required Methods

fn get_id(&self) -> String

fn get_dim(&self) -> Vec2

fn run(&self, state: &Context<T>, variable: &Context<T>) -> Tensor<T>

fn forward_pass(
    &self,
    state: &Context<T>,
    variable: &Context<T>,
    history: &mut Context<T>
) -> Tensor<T>

fn backward_pass(
    &self,
    state: &mut Context<T>,
    variable: &Context<T>,
    history: &Context<T>,
    gradient: &Tensor<T>,
    learning_rate: T
)

Provided Methods

fn train(
    &self,
    state: &Context<T>,
    variable: &Context<T>,
    history: &mut Context<T>
) -> Tensor<T>

Implementors

• impl<T> Graph<T> for Node<T> where
    T: Copy + Mul<Output = T> + Add<Output = T>, 
• impl<T> Graph<T> for State where
    T: Copy + Mul<Output = T> + Add<Output = T>, 
• impl<T> Graph<T> for Variable where
    T: Copy + Mul<Output = T> + Add<Output = T>,