Trait Module

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

Defining a module with forward method using a single argument.

Required Methods

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

Implementations on Foreign Types

impl<M> Module for Option<&M>

where M: Module,

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

Implementors

impl Module for Activation

impl Module for QMatMul

impl Module for PReLU

impl Module for Conv1d

impl Module for Conv2d

impl Module for ConvTranspose1d

impl Module for ConvTranspose2d

impl Module for Embedding

impl Module for Func<'_>

impl Module for GroupNorm

impl Module for LayerNorm

impl Module for RmsNorm

impl Module for Linear

impl Module for Identity

impl Module for Sequential

impl<T> Module for T

where T: Fn(&Tensor) -> Result<Tensor, Error>,