Trait Layer 

pub trait Layer {
    // Required methods
    fn parameters(&self) -> Vec<&Tensor>;
    fn forward(&self, x: &Tensor) -> Tensor ;
    fn shape(&self) -> (usize, usize);

    // Provided method
    fn zero_grad(&self) { ... }
}

A general trait of a layer of a neural network

Required Methods

fn parameters(&self) -> Vec<&Tensor>

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

fn shape(&self) -> (usize, usize)

Provided Methods

fn zero_grad(&self)

Trait Implementations

impl Debug for dyn Layer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

Implementors

impl Layer for Linear