package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-05 07:06:54 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

optimize: avoid unnecessary allocations for Gradient and Hessian

Browse Source
This commit is contained in:
Dan Kortschak
2020-02-16 15:45:16 +10:30
parent 7d9e94571f
commit 8ad895b51b
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
optimize/minimize.go
View File

@@ -478,19 +478,22 @@ func evaluate(p *Problem, loc *Location, op Operation, x []float64) {
}
if op&GradEvaluation != 0 {
// Make sure we have a destination in which to place the gradient.
// TODO(kortschak): Consider making this a check of len(loc.Gradient) != 0
// to allow reuse of the slice.
if loc.Gradient == nil {
loc.Gradient = make([]float64, len(x))
if len(loc.Gradient) == 0 {
if cap(loc.Gradient) < len(x) {
loc.Gradient = make([]float64, len(x))
} else {
loc.Gradient = loc.Gradient[:len(x)]
}
}
p.Grad(loc.Gradient, x)
}
if op&HessEvaluation != 0 {
// Make sure we have a destination in which to place the Hessian.
// TODO(kortschak): Consider making this a check of loc.Hessian.IsZero()
// to allow reuse of the matrix.
if loc.Hessian == nil {
switch {
case loc.Hessian == nil:
loc.Hessian = mat.NewSymDense(len(x), nil)
case loc.Hessian.IsEmpty():
loc.Hessian.ReuseAsSym(len(x))
}
p.Hess(loc.Hessian, x)
}