Skip to main content

Module

Trait Module 

Source
pub trait Module<I = &'static Tensor> {
    // Required methods
    fn forward(&self, input: I) -> Result<Tensor, HoduError>;
    fn parameters(&mut self) -> Vec<&mut Tensor>;
}

Required Methods§

Source

fn forward(&self, input: I) -> Result<Tensor, HoduError>

Source

fn parameters(&mut self) -> Vec<&mut Tensor>

Dyn Compatibility§

This trait is dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors§

Source§

impl Module<&Tensor> for AdaptiveAvgPool1D

Source§

impl Module<&Tensor> for AdaptiveAvgPool2D

Source§

impl Module<&Tensor> for AdaptiveAvgPool3D

Source§

impl Module<&Tensor> for AdaptiveMaxPool1D

Source§

impl Module<&Tensor> for AdaptiveMaxPool2D

Source§

impl Module<&Tensor> for AdaptiveMaxPool3D

Source§

impl Module<&Tensor> for AvgPool1D

Source§

impl Module<&Tensor> for AvgPool2D

Source§

impl Module<&Tensor> for AvgPool3D

Source§

impl Module<&Tensor> for BatchNorm1D

Source§

impl Module<&Tensor> for BatchNorm2D

Source§

impl Module<&Tensor> for BatchNorm3D

Source§

impl Module<&Tensor> for Conv1D

Source§

impl Module<&Tensor> for Conv2D

Source§

impl Module<&Tensor> for Conv3D

Source§

impl Module<&Tensor> for ConvTranspose1D

Source§

impl Module<&Tensor> for ConvTranspose2D

Source§

impl Module<&Tensor> for ConvTranspose3D

Source§

impl Module<&Tensor> for Dropout

Source§

impl Module<&Tensor> for ELU

Source§

impl Module<&Tensor> for Embedding

Source§

impl Module<&Tensor> for Gelu

Source§

impl Module<&Tensor> for LayerNorm

Source§

impl Module<&Tensor> for LeakyReLU

Source§

impl Module<&Tensor> for Linear

Source§

impl Module<&Tensor> for MaxPool1D

Source§

impl Module<&Tensor> for MaxPool2D

Source§

impl Module<&Tensor> for MaxPool3D

Source§

impl Module<&Tensor> for Mish

Source§

impl Module<&Tensor> for PReLU

Source§

impl Module<&Tensor> for RReLU

Source§

impl Module<&Tensor> for ReLU

Source§

impl Module<&Tensor> for SiLU

Source§

impl Module<&Tensor> for Sigmoid

Source§

impl Module<&Tensor> for Softplus

Source§

impl Module<&Tensor> for Swish

Source§

impl Module<&Tensor> for Tanh

Source§

impl Module<(&Tensor, &Tensor)> for BCELoss

Source§

impl Module<(&Tensor, &Tensor)> for BCEWithLogitsLoss

Source§

impl Module<(&Tensor, &Tensor)> for CrossEntropyLoss

Source§

impl Module<(&Tensor, &Tensor)> for HuberLoss

Source§

impl Module<(&Tensor, &Tensor)> for MAELoss

Source§

impl Module<(&Tensor, &Tensor)> for MSELoss

Source§

impl Module<(&Tensor, &Tensor)> for NLLLoss