Trait tch::nn::Module

pub trait Module: Debug + Send {
    fn forward(&self, xs: &Tensor) -> Tensor;
}

The simplest module trait, defining a forward function.

Required methods

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

Implementors

impl Module for CModule

impl Module for Embedding

impl Module for LayerNorm

impl Module for Linear

impl Module for Sequential

impl Module for Conv1D

impl Module for Conv2D

impl Module for Conv3D

impl Module for ConvTranspose1D

impl Module for ConvTranspose2D

impl Module for ConvTranspose3D

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