descent

Documentation

Index

• type Descent
  • func MustNew(eta float64) *Descent
  • func New(eta float64) (*Descent, error)
  • func (o *Descent) Extend(_ int)
  • func (o *Descent) Update(out, p *mat.VecDense, g mat.Vector) *mat.VecDense

Types

type Descent

type Descent struct {
	// contains filtered or unexported fields
}

Gradient descent optimiser; see the following link for details: https://fluxml.ai/Flux.jl/stable/training/optimisers/

func MustNew

func MustNew(eta float64) *Descent

func New

func New(eta float64) (*Descent, error)

func (*Descent) Extend

func (o *Descent) Extend(_ int)

func (*Descent) Update

func (o *Descent) Update(out, p *mat.VecDense, g mat.Vector) *mat.VecDense