public class RiemannSGD<Model: Layer, Scalar: FloatingPoint>: Optimizer
    where Model.TangentVector: VectorNumeric, Model.TangentVector.Scalar == Scalar

A Riemann manifold stochastic gradient descent (SGD) optimizer.

  • The learning rate.

    Declaration

    public var learningRate: Scalar
  • Declaration

    public init(learningRate: Scalar)
  • Declaration

    public convenience init(
        for _: __shared Model,
        learningRate: Scalar
    )
  • Declaration

    public func update(_ model: inout Model.AllDifferentiableVariables,
                       along direction: Model.CotangentVector)