Skip to main content

Module

tch_plus::nn

Trait Module 

Source 
pub trait Module: Debug + Send {
    // Required method
    fn forward(&self, xs: &Tensor) -> Tensor;
}
Expand description

The simplest module trait, defining a forward function.

Required Methods§

Source

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

Dyn Compatibility§

This trait is dyn compatible.

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

Implementors§