Enum GraphOp 

pub enum GraphOp {
    Constant,
    Add,
    Mul,
    Sub,
    Div,
    MatMul,
    Reshape(Vec<usize>),
    Transpose(usize, usize),
    Sum(Option<i32>),
    Mean(Option<i32>),
    ReLU,
    Sigmoid,
    Tanh,
    AddScalar(f64),
    MulScalar(f64),
    Custom(String),
}

Operation types in the computation graph

Variants

Constant

Input/constant tensor

Add

Element-wise addition

Mul

Element-wise multiplication

Sub

Element-wise subtraction

Div

Element-wise division

MatMul

Matrix multiplication

Reshape(Vec<usize>)

Reshape operation

Transpose(usize, usize)

Transpose operation

Sum(Option<i32>)

Reduction sum

Mean(Option<i32>)

Reduction mean

ReLU

ReLU activation

Sigmoid

Sigmoid activation

Tanh

Tanh activation

AddScalar(f64)

Scalar addition

MulScalar(f64)

Scalar multiplication

Custom(String)

Custom operation

Trait Implementations

impl Clone for GraphOp

fn clone(&self) -> GraphOp

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GraphOp

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

Formats the value using the given formatter.

impl Display for GraphOp

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

Formats the value using the given formatter.