Struct auto_diff::var::Var

[+] Show declaration

pub struct Var { /* fields omitted */ }

[−] Expand description

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[+]

[+] Show hidden undocumented items

fn fmt(&self, f: &mut Formatter) -> Result[−]

Formats the value using the given formatter.

impl Drop for Var[+]

[+] Show hidden undocumented items

fn drop(&mut self)[−]

Executes the destructor for this type.