Trait TensorOp 

pub trait TensorOp: Send + Sync {
    // Required methods
    fn name(&self) -> &str;
    fn execute(&self, input: &Tensor) -> Result<Tensor, TribeError>;
}

Trait for all tensor operations (aligned with .AI3 TensorOp).

Required Methods

fn name(&self) -> &str

fn execute(&self, input: &Tensor) -> Result<Tensor, TribeError>

Implementors

impl TensorOp for ActivationFunction

impl TensorOp for VectorOp

impl TensorOp for Convolution

impl TensorOp for MatrixMultiply