Trait candle_nn::Module

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

Required Methods

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 GroupNorm

impl Module for LayerNorm

impl Module for RmsNorm

impl Module for Linear

impl Module for Sequential

impl<'a> Module for Func<'a>

impl<T> Module for T

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