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

Loading content...

Implementors

impl Module for CModule

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

impl Module for Embedding

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

impl Module for LayerNorm

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

impl Module for Linear

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

impl Module for Sequential

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

impl Module for Conv1D

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

impl Module for Conv2D

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

impl Module for Conv3D

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

impl Module for ConvTranspose1D

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

impl Module for ConvTranspose2D

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

impl Module for ConvTranspose3D

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

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

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

Loading content...