Trait yarnn::optimizer::OptimizerContext

pub trait OptimizerContext {
    fn new<S: Into<TensorShape>>(shape: S) -> Self;
}

Required methods

fn new<S: Into<TensorShape>>(shape: S) -> Self

Loading content...

Implementors

impl<N, B: Backend<N>> OptimizerContext for AdamContext<N, B>

fn new<S: Into<TensorShape>>(shape: S) -> Self

impl<N, B: Backend<N>> OptimizerContext for RMSPropContext<N, B>

fn new<S: Into<TensorShape>>(shape: S) -> Self

impl<N, B: Backend<N>> OptimizerContext for SgdContext<N, B>

fn new<S: Into<TensorShape>>(shape: S) -> Self

Loading content...