File size: 226 Bytes
e9fe176
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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)