Struct SafeModule 

pub struct SafeModule<M, S, E> { /* private fields */ }

Implementations

impl<M, S, E> SafeModule<M, S, E>

pub fn into_inner(self) -> M

impl<M: Module> SafeModule<M, Forward, Train>

pub fn new(module: M) -> Self

impl<M: Module, E> SafeModule<M, Forward, E>

pub fn forward( self, input: Array2<f64>, ) -> (SafeModule<M, Backward, E>, Array2<f64>)

(batch_size, input_size) -> (batch_size, output_size)

impl<M: Module, E> SafeModule<M, Backward, E>

pub fn backward( self, gradient: Array2<f64>, ) -> (SafeModule<M, Forward, E>, Array2<f64>)

(batch_size, output_size) -> (batch_size, input_size)

impl<M: Module, S> SafeModule<M, S, Train>

pub fn eval(self) -> SafeModule<M, S, Evaluation>

impl<M: Module, S> SafeModule<M, S, Evaluation>

pub fn train(self) -> SafeModule<M, S, Train>

Trait Implementations

impl<M: Clone, S: Clone, E: Clone> Clone for SafeModule<M, S, E>

fn clone(&self) -> SafeModule<M, S, E>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<M: Debug, S: Debug, E: Debug> Debug for SafeModule<M, S, E>

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

Formats the value using the given formatter.

impl<M: Default, S: Default, E: Default> Default for SafeModule<M, S, E>

fn default() -> SafeModule<M, S, E>

Returns the “default value” for a type.

impl<M: Module> From<M> for SafeModule<M, Forward, Train>

fn from(module: M) -> Self

Converts to this type from the input type.

impl<M: Hash, S: Hash, E: Hash> Hash for SafeModule<M, S, E>

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl<M: Ord, S: Ord, E: Ord> Ord for SafeModule<M, S, E>

fn cmp(&self, other: &SafeModule<M, S, E>) -> Ordering

This method returns an Ordering between self and other.

fn max(self, other: Self) -> Self

where Self: Sized,

Compares and returns the maximum of two values.

fn min(self, other: Self) -> Self

where Self: Sized,

Compares and returns the minimum of two values.

fn clamp(self, min: Self, max: Self) -> Self

where Self: Sized,

Restrict a value to a certain interval.

impl<M: PartialEq, S: PartialEq, E: PartialEq> PartialEq for SafeModule<M, S, E>

fn eq(&self, other: &SafeModule<M, S, E>) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl<M: PartialOrd, S: PartialOrd, E: PartialOrd> PartialOrd for SafeModule<M, S, E>

fn partial_cmp(&self, other: &SafeModule<M, S, E>) -> Option<Ordering>

This method returns an ordering between self and other values if one exists.

fn lt(&self, other: &Rhs) -> bool

Tests less than (for self and other) and is used by the < operator.

fn le(&self, other: &Rhs) -> bool

Tests less than or equal to (for self and other) and is used by the <= operator.

fn gt(&self, other: &Rhs) -> bool

Tests greater than (for self and other) and is used by the > operator.

fn ge(&self, other: &Rhs) -> bool

Tests greater than or equal to (for self and other) and is used by the >= operator.

impl<M: Copy, S: Copy, E: Copy> Copy for SafeModule<M, S, E>

impl<M: Eq, S: Eq, E: Eq> Eq for SafeModule<M, S, E>

impl<M, S, E> StructuralPartialEq for SafeModule<M, S, E>