Trait Parameters

pub trait Parameters<T: Num, D: Device> {
    // Required methods
    fn weights(&self) -> HashMap<String, Variable<T, D>>;
    fn biases(&self) -> HashMap<String, Variable<T, D>>;

    // Provided methods
    fn parameters(&self) -> HashMap<String, Variable<T, D>> { ... }
    fn load_parameters(&mut self, parameters: HashMap<String, Variable<T, D>>) { ... }
}

Required Methods

fn weights(&self) -> HashMap<String, Variable<T, D>>

fn biases(&self) -> HashMap<String, Variable<T, D>>

Provided Methods

fn parameters(&self) -> HashMap<String, Variable<T, D>>

fn load_parameters(&mut self, parameters: HashMap<String, Variable<T, D>>)

Trait Implementations

impl<T: Num, D: Device> Parameters<T, D> for Box<dyn Parameters<T, D>>

fn weights(&self) -> HashMap<String, Variable<T, D>>

fn biases(&self) -> HashMap<String, Variable<T, D>>

fn parameters(&self) -> HashMap<String, Variable<T, D>>

fn load_parameters(&mut self, parameters: HashMap<String, Variable<T, D>>)

Implementations on Foreign Types

impl<T: Num, D: Device> Parameters<T, D> for ()

fn weights(&self) -> HashMap<String, Variable<T, D>>

fn biases(&self) -> HashMap<String, Variable<T, D>>

impl<T: Num, D: Device> Parameters<T, D> for Box<dyn Parameters<T, D>>

fn weights(&self) -> HashMap<String, Variable<T, D>>

fn biases(&self) -> HashMap<String, Variable<T, D>>

impl<T: Num, D: Device, P: Parameters<T, D>> Parameters<T, D> for Vec<P>

fn weights(&self) -> HashMap<String, Variable<T, D>>

fn biases(&self) -> HashMap<String, Variable<T, D>>

impl<T: Num, D: Device, P: Parameters<T, D>, S: BuildHasher> Parameters<T, D> for HashMap<String, P, S>

fn weights(&self) -> HashMap<String, Variable<T, D>>

fn biases(&self) -> HashMap<String, Variable<T, D>>

Implementors

impl<T: Num, D: Device> Parameters<T, D> for BatchNorm2d<T, D>

impl<T: Num, D: Device> Parameters<T, D> for Conv2d<T, D>

impl<T: Num, D: Device> Parameters<T, D> for Dropout<T, D>

impl<T: Num, D: Device> Parameters<T, D> for Linear<T, D>

impl<T: Num, D: Device> Parameters<T, D> for MaxPool2d<T>

impl<T: Num, D: Device> Parameters<T, D> for GRU<T, D>

impl<T: Num, D: Device> Parameters<T, D> for LSTM<T, D>

impl<T: Num, D: Device> Parameters<T, D> for RNN<T, D>