Trait Module

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

Defining a module with forward method using a single argument.

Required Methods

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

Implementations on Foreign Types

impl<M: Module> Module for Option<&M>

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

Implementors

impl Module for QMatMul

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