Struct DynamicTensor 

pub struct DynamicTensor {
    pub tensor: Tensor,
    pub node_id: Option<usize>,
    pub graph: Option<Arc<DynamicGraph>>,
}

Wrapper for tensors in dynamic graph

Fields

tensor: Tensor

The actual tensor

node_id: Option<usize>

Node ID in the graph

graph: Option<Arc<DynamicGraph>>

Reference to the graph

Implementations

impl DynamicTensor

pub fn new(tensor: Tensor, graph: Arc<DynamicGraph>) -> Self

Create a new dynamic tensor

pub fn from_tensor(tensor: Tensor) -> Self

Create from tensor without graph

pub fn add(&self, other: &DynamicTensor) -> DynamicTensor

Add two dynamic tensors

pub fn mul(&self, other: &DynamicTensor) -> DynamicTensor

Multiply two dynamic tensors

pub fn matmul(&self, other: &DynamicTensor) -> DynamicTensor

Matrix multiplication

pub fn relu(&self) -> DynamicTensor

ReLU activation

pub fn backward(&self) -> HashMap<usize, Tensor>

Compute gradients

Trait Implementations

impl Clone for DynamicTensor

fn clone(&self) -> DynamicTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DynamicTensor

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

Formats the value using the given formatter.