Trait rai_core::nn::WithParams

pub trait WithParams {
    // Required methods
    fn gather_by_id(&self, params: &mut HashMap<usize, Tensor>);
    fn update_by_id(&self, params: &mut HashMap<usize, Tensor>);
    fn gather_by_name(
        &self,
        params: &mut HashMap<String, Tensor>,
        prefix: &str,
        name: &str
    );
    fn update_by_name(
        &self,
        params: &mut HashMap<String, Tensor>,
        prefix: &str,
        name: &str
    );
}

Required Methods

fn gather_by_id(&self, params: &mut HashMap<usize, Tensor>)

fn update_by_id(&self, params: &mut HashMap<usize, Tensor>)

fn gather_by_name( &self, params: &mut HashMap<String, Tensor>, prefix: &str, name: &str )

fn update_by_name( &self, params: &mut HashMap<String, Tensor>, prefix: &str, name: &str )

Implementations on Foreign Types

impl<T> WithParams for Option<T>

where T: WithParams,

fn gather_by_id(&self, params: &mut HashMap<usize, Tensor>)

fn update_by_id(&self, params: &mut HashMap<usize, Tensor>)

fn gather_by_name( &self, params: &mut HashMap<String, Tensor>, prefix: &str, name: &str )

fn update_by_name( &self, params: &mut HashMap<String, Tensor>, prefix: &str, name: &str )

impl<T> WithParams for Vec<T>

where T: WithParams,

fn gather_by_id(&self, params: &mut HashMap<usize, Tensor>)

fn update_by_id(&self, params: &mut HashMap<usize, Tensor>)

fn gather_by_name( &self, params: &mut HashMap<String, Tensor>, prefix: &str, name: &str )

fn update_by_name( &self, params: &mut HashMap<String, Tensor>, prefix: &str, name: &str )

Implementors

impl WithParams for Tensor

impl<T> WithParams for T

where T: Module,