Struct GradientDescentOptimizer

pub struct GradientDescentOptimizer { /* private fields */ }

Optimizer that implements the gradient descent algorithm.

Implementations

impl GradientDescentOptimizer

pub fn new<T: Into<Output>>(learning_rate: T) -> Self

Creates a new optimizer with the given learning rate.

Trait Implementations

impl Debug for GradientDescentOptimizer

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

Formats the value using the given formatter.

impl Optimizer for GradientDescentOptimizer

fn apply_gradients( &self, scope: &mut Scope, opts: ApplyGradientsOptions<'_>, ) -> Result<(Vec<Variable>, Operation)>

Applies the given gradients to the variables.

fn compute_gradients( &self, scope: &mut Scope, loss: Output, opts: ComputeGradientsOptions<'_>, ) -> Result<Vec<(Option<Output>, Variable)>>

Computes the gradient of a value with respect to the given variables. This adds nodes to the graph, so reuse its results if possible. Any variable whose gradient cannot be calculated will have a None gradient.

fn minimize( &self, scope: &mut Scope, loss: Output, opts: MinimizeOptions<'_>, ) -> Result<(Vec<Variable>, Operation)>

Adds operations to the graph to minimize loss with respect to the variables.