FloorPlanTransformation / util /lua /nn /test_autograd.lua
rawanessam's picture
Upload 79 files
e9fe176 verified
require 'torch'
grad = require 'autograd'
function net(params, x)
local p = params[1][2]
local q = params[1][3]
return p + 2 * q
end
dnet = grad(net)
params = torch.Tensor({{2, 3, 3}})
dparams = dnet(params, x)