Trait ModuleT 

pub trait ModuleT {
    // Required method
    fn forward_t(&self, xs: &Tensor, train: bool) -> Result<Tensor, Error>;
}

A single forward method using a single single tensor argument and a flag to separate the training and evaluation behaviors.

Required Methods

fn forward_t(&self, xs: &Tensor, train: bool) -> Result<Tensor, Error>

Dyn Compatibility

This trait is dyn compatible.

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

Implementors

impl ModuleT for BatchNorm

impl ModuleT for Dropout

impl ModuleT for FuncT<'_>

impl<M> ModuleT for M

where M: Module,