标签: mxnet.jl

如何使用mxnet.jl和Julia获取渐变节点?

我试图在Julia中使用mxnet.jl从mxnet主文档复制以下示例:

A = Variable('A')
B = Variable('B')
C = B * A
D = C + Constant(1)
# get gradient node.
gA, gB = D.grad(wrt=[A, B])
# compiles the gradient function.
f = compile([gA, gB])
grad_a, grad_b = f(A=np.ones(10), B=np.ones(10)*2)
Run Code Online (Sandbox Code Playgroud)

该示例显示如何自动提取symoblic表达式并获取其渐变.

mxnet.jl(2016-03-07最新版本)中的等价物是什么?

gradient-descent julia mxnet mxnet.jl

5
推荐指数
1
解决办法
279
查看次数

标签 统计

gradient-descent ×1

julia ×1

mxnet ×1

mxnet.jl ×1