| 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) | |
| 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) | |