Struct ManifoldRGD 

pub struct ManifoldRGD<M: Manifold<B>, B: Backend> { /* private fields */ }

Trait Implementations

impl<M: Clone + Manifold<B>, B: Clone + Backend> Clone for ManifoldRGD<M, B>

fn clone(&self) -> ManifoldRGD<M, B>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<M: Debug + Manifold<B>, B: Debug + Backend> Debug for ManifoldRGD<M, B>

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

Formats the value using the given formatter.

impl<M, B> Default for ManifoldRGD<M, B>

where M: Manifold<B>, B: Backend,

fn default() -> Self

Returns the “default value” for a type.

impl<M, B> SimpleOptimizer<B> for ManifoldRGD<M, B>

where M: Manifold<B>, B: Backend,

type State<const D: usize> = ManifoldRGDState<B, D>

The state of the optimizer. It also implements record, so that it can be saved.

fn step<const D: usize>( &self, lr: LearningRate, tensor: Tensor<B, D>, grad: Tensor<B, D>, state: Option<Self::State<D>>, ) -> (Tensor<B, D>, Option<Self::State<D>>)

The optimizer step is performed for one tensor at a time with its gradient and state.

fn to_device<const D: usize>( _state: Self::State<D>, _device: &<B as Backend>::Device, ) -> Self::State<D>

Change the device of the state.