Struct ktensor::Variable

pub struct Variable { /* fields omitted */ }

Methods

impl Variable

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

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_variables: Vec<Arc<Variable>>, context: &mut Context<f64>)

fn init_f32(vec_variables: Vec<Arc<Variable>>, context: &mut Context<f32>)

Trait Implementations

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

fn get_id(&self) -> String

fn get_dim(&self) -> Vec2

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

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

fn backward_pass(
    &self,
    _: &mut Context<T>,
    _: &Context<T>,
    _: &Context<T>,
    _: &Tensor<T>,
    _: T
)

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