Struct rai_core::nn::Chain

pub struct Chain<A, B> { /* private fields */ }

Implementations

impl<A, B> Chain<A, B>

pub fn new(a: A, b: B) -> Self

Trait Implementations

impl<A, B, T> Module for Chain<A, B>

where A: Module<Input = T>, B: Module<Input = A::Output>,

type Input = T

type Output = <B as Module>::Output

fn forward(&self, x: &Self::Input) -> Self::Output

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

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

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

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

fn params(&self) -> HashMap<usize, Tensor>

fn named_params(&self, prefix: &str) -> HashMap<String, Tensor>

fn to_safetensors<P: AsRef<Path>>(&self, filename: P)

where Self: Sized,

fn update_by_safetensors<P: AsRef<Path>>(&self, filenames: &[P])

fn chain<B>(self, b: B) -> Chain<Self, B>

where Self: Sized,