Как получить градиентный узел с mxnet.jl и Джулией?

Question

Я пытаюсь повторить следующий пример из основной документации mxnet с mxnet.jl в Юлии:

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)

В примере показано, как автодифференцировать символическое выражение и получить его градиенты.

Что является эквивалентом в mxnet.jl (последняя версия 2016-03-07)?

Answer 1

Код в MXNet.jl/src/symbolic-node.jl может быть полезно для вас найти ответы.

Я не знаком с этим пакетом. Вот моя догадка: A = mx.Variable("A") B = mx.Variable("B") C = B .* A D = C + 1 mx.normalized_gradient может быть решением оставшейся части, если существует.