2016-03-07 12 views
5

Tôi đang cố gắng để nhân rộng các ví dụ sau đây từ các tài liệu chính mxnet với mxnet.jl trong Julia:Làm thế nào để có được một nút gradient với mxnet.jl và Julia?

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) 

Ví dụ cho thấy làm thế nào để autodiff một biểu symoblic và có được độ dốc của nó.

Tương đương trong mxnet.jl (phiên bản mới nhất 2016-03-07) là gì?

Trả lời

1

Mã số trong MXNet.jl/src/symbolic-node.jl có thể hữu ích để bạn tìm câu trả lời.

Tôi không quen với gói này. Đây là Guess của tôi: A = mx.Variable("A") B = mx.Variable("B") C = B .* A D = C + 1 mx.normalized_gradient có thể là giải pháp cho phần còn lại nếu tồn tại.

Các vấn đề liên quan