Struct TracedValue 

pub struct TracedValue { /* private fields */ }

A traced value representing a node in the computation graph.

Implementations

impl TracedValue

pub fn node_id(&self) -> NodeId

Returns the node ID.

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

Element-wise addition.

pub fn sub(&self, other: &TracedValue) -> TracedValue

Element-wise subtraction.

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

Element-wise multiplication.

pub fn div(&self, other: &TracedValue) -> TracedValue

Element-wise division.

pub fn pow(&self, exp: &TracedValue) -> TracedValue

Element-wise power.

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

Matrix multiplication.

pub fn add_scalar(&self, scalar: f64) -> TracedValue

Add scalar.

pub fn mul_scalar(&self, scalar: f64) -> TracedValue

Multiply by scalar.

pub fn neg(&self) -> TracedValue

Negation.

pub fn abs(&self) -> TracedValue

Absolute value.

pub fn sqrt(&self) -> TracedValue

Square root.

pub fn exp(&self) -> TracedValue

Exponential.

pub fn log(&self) -> TracedValue

Natural logarithm.

pub fn sin(&self) -> TracedValue

Sine.

pub fn cos(&self) -> TracedValue

Cosine.

pub fn tanh(&self) -> TracedValue

Hyperbolic tangent.

pub fn relu(&self) -> TracedValue

ReLU activation.

pub fn sigmoid(&self) -> TracedValue

Sigmoid activation.

pub fn gelu(&self) -> TracedValue

GELU activation.

pub fn silu(&self) -> TracedValue

SiLU/Swish activation.

pub fn sum(&self) -> TracedValue

Sum over all elements.

pub fn sum_axis(&self, axis: i32, keepdim: bool) -> TracedValue

Sum over axis.

pub fn mean(&self) -> TracedValue

Mean over all elements.

pub fn mean_axis(&self, axis: i32, keepdim: bool) -> TracedValue

Mean over axis.

pub fn reshape(&self, shape: &[isize]) -> TracedValue

Reshape tensor.

pub fn transpose(&self, dim0: usize, dim1: usize) -> TracedValue

Transpose dimensions.

pub fn squeeze(&self, dim: i32) -> TracedValue

Squeeze dimension.

pub fn unsqueeze(&self, dim: i32) -> TracedValue

Unsqueeze (add dimension).

Trait Implementations

impl Clone for TracedValue

fn clone(&self) -> TracedValue

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TracedValue

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

Formats the value using the given formatter.

impl Copy for TracedValue