Struct TensorExpr 

pub struct TensorExpr { /* private fields */ }

Tensor expression for lazy evaluation

Implementations

impl TensorExpr

pub fn from(tensor: &Tensor) -> Result<Self>

Create a new expression from a tensor

pub fn constant(tensor: Tensor) -> Result<Self>

Create a constant expression

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

Get the shape of the expression result

pub fn dtype(&self) -> DType

Get the data type of the expression result

pub fn add(self, other: TensorExpr) -> Result<Self>

Add two expressions

pub fn sub(self, other: TensorExpr) -> Result<Self>

Subtract two expressions

pub fn mul(self, other: TensorExpr) -> Result<Self>

Multiply two expressions element-wise

pub fn div(self, other: TensorExpr) -> Result<Self>

Divide two expressions element-wise

pub fn matmul(self, other: TensorExpr) -> Result<Self>

Matrix multiplication

pub fn relu(self) -> Result<Self>

Apply ReLU activation

pub fn sigmoid(self) -> Result<Self>

Apply sigmoid activation

pub fn tanh(self) -> Result<Self>

Apply tanh activation

pub fn gelu(self) -> Result<Self>

Apply GELU activation

pub fn softmax(self, axis: i32) -> Result<Self>

Apply softmax along the specified axis

pub fn sum(self, axes: Option<Vec<usize>>) -> Result<Self>

Sum along specified axes

pub fn mean(self, axes: Option<Vec<usize>>) -> Result<Self>

Calculate mean along specified axes

pub fn reshape(self, shape: &[usize]) -> Result<Self>

Reshape the tensor

pub fn transpose(self) -> Result<Self>

Transpose the tensor

pub fn eval(&self) -> Result<Tensor>

Evaluate the expression with default context

pub fn eval_with_context(&self, context: &EvalContext) -> Result<Tensor>

Evaluate the expression with optimization context

pub fn can_fuse_with(&self, other: &TensorExpr) -> bool

Check if two expressions can be fused

pub fn operation_count(&self) -> usize

Get the number of operations in the expression

pub fn leaf_count(&self) -> usize

Get the number of leaf nodes (tensors)

pub fn to_dot(&self) -> String

Export expression to DOT format for visualization

Trait Implementations

impl Clone for TensorExpr

fn clone(&self) -> TensorExpr

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorExpr

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for TensorExpr

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for TensorExpr

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

Formats the value using the given formatter.

impl Serialize for TensorExpr

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.