Trait tch::nn::Module

pub trait Module: Debug {
    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 ConvTranspose2D

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<'a> Module for Func<'a>

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

Loading content...