1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
require 'torch'
-- rosenbrock.m This function returns the function value, partial derivatives
-- and Hessian of the (general dimension) rosenbrock function, given by:
--
-- f(x) = sum_{i=1:D-1} 100*(x(i+1) - x(i)^2)^2 + (1-x(i))^2
--
-- where D is the dimension of x. The true minimum is 0 at x = (1 1 ... 1).
--
-- Carl Edward Rasmussen, 2001-07-21.
function l2(x)
local xx = x:clone()
xx:cmul(xx)
local fout = xx:sum()
local dx = torch.Tensor():resizeAs(x):copy(x)
dx:mul(2)
--print('l2 eval = ', fout)
return fout,dx
end
|
