Struct GradientTensor 

pub struct GradientTensor { /* private fields */ }

Tensor with gradient tracking capabilities.

Implementations

impl GradientTensor

pub fn new(value: Rc<dyn ArrayProtocol>, requiresgrad: bool) -> Self

Create a new gradient tensor from a value.

pub fn from_array<T, D>(array: Array<T, D>, requiresgrad: bool) -> Self

where T: Clone + Send + Sync + 'static, D: Dimension + Send + Sync + 'static,

Create a new gradient tensor from an array.

pub fn value(&self) -> Rc<dyn ArrayProtocol>

Get the value of the tensor.

pub fn grad_2(&self) -> Option<Rc<dyn ArrayProtocol>>

Get the gradient of the tensor.

pub fn requiresgrad(&self) -> bool

Check if gradient computation is required for this tensor.

pub fn set_requiresgrad(&mut self, requiresgrad: bool)

Set whether gradient computation is required for this tensor.

pub fn is_leaf(&self) -> bool

Check if this tensor is a leaf node.

pub fn set_value(&mut self, newvalue: Rc<dyn ArrayProtocol>)

Set the value of the tensor (for updating variables during optimization).

pub fn backward(&self) -> CoreResult<()>

Backward pass to compute gradients.

pub fn detach(&self) -> Self

Detach this tensor from the computation graph.

Trait Implementations

impl Clone for GradientTensor

fn clone(&self) -> GradientTensor

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.