Trait Module

pub trait Module {
    // Required method
    fn forward(&self, xs: &Tensor) -> Result<Tensor>;
}

Required Methods

fn forward(&self, xs: &Tensor) -> Result<Tensor>

Implementors

impl Module for QMatMul

impl<T: Fn(&Tensor) -> Result<Tensor>> Module for T