Struct ktensor::State

pub struct State { /* fields omitted */ }

Methods

impl State

fn new(node_id: String, dimensions: Vec2) -> State

fn get_id(&self) -> String

fn init_norm_f64(&self, context: &mut Context<f64>)

fn init_norm_f32(&self, context: &mut Context<f32>)

fn init_f64(vec_states: Vec<Arc<State>>, context: &mut Context<f64>)

fn init_f32(vec_states: Vec<Arc<State>>, context: &mut Context<f32>)

Trait Implementations

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

fn get_id(&self) -> String

fn get_dim(&self) -> Vec2

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

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

fn backward_pass(
    &self,
    state: &mut Context<T>,
    _: &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>