Struct auto_diff::var::Var

pub struct Var { /* private fields */ }

Introduce variable to the system by creating Var

Implementations

impl Var

pub fn _default() -> Var

pub fn _new(id: NetIndex, net: Rc<RefCell<Net>>) -> Var

pub fn new_attached(&self) -> Var

Create a variable using an existing variable.

pub fn new_variable(net: Rc<RefCell<Net>>) -> Var

Create a variable given the network. This is for whoever has a network handler.

pub fn get_id(&self) -> &NetIndex

Getter for the id member.

pub fn set_id(&mut self, new_id: NetIndex)

Setter for the id member.

pub fn set(&self, v: Tensor)

Give the variable a value

let mut m = Module::new();
let a = m.var();
a.set(Tensor::new());

pub fn unset(&self)

pub fn get(&self) -> Tensor

Get the underlying tensor.

pub fn get_grad(&self) -> Tensor

Get the underlying gradient tensor.

pub fn to(&self, op: &Op) -> Var

apply the var to pre-faburacated op.

pub fn backward(&self, og: f32)

pub fn size(&self) -> Vec<usize>

pub fn numel(&self) -> usize

pub fn add(&self, o: &Var) -> Var

pub fn sub(&self, o: &Var) -> Var

pub fn mul(&self, o: &Var) -> Var

pub fn div(&self, o: &Var) -> Var

Trait Implementations

impl Display for Var

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Drop for Var

fn drop(&mut self)

Executes the destructor for this type.