Mxnet元素明智乘法

Question

在MXNet中，如果我想創建一個權重向量，使其與每個輸入相乘，即具有w*x_i，然後在權重w上反向傳遞，我該怎麼做？

我想：

y_hat = input 
w1 = mx.sym.Variable("w1") 
y_hat = mx.symbol.broadcast_mul(w1, y_hat) 

Answer 1

，就可以把計算的點積方面：

x = mx.nd.array([[1, 2, 3], [4, 5, 6]]) 
w = mx.nd.array([2,2,2]) 
mx.nd.dot(w, x.T) 

將導致[12 30]如你所願。

現在只需隨機初始化w，計算輸出和目標輸出之間的損失，然後再傳播。您可以使用新的gluon接口（http://gluon.mxnet.io/）。

具體，讓我們來看看一個小例子，適合http://mxnet.io/tutorials/gluon/gluon.html和http://gluon.mxnet.io/P01-C05-autograd.html

準備數據

label = mx.nd.array([12,30]) 
x = mx.nd.array([[1, 2, 3], [4, 5, 6]]) 
w = random weights 
w.attach_grad() 

和火車站

with autograd.record(): 
    output = mx.nd.dot(w, x.T) 
    loss = gluon.loss.L2Loss(output, label) 
    loss.backward() 

不要用你所計算的梯度忘記updating the weights落後的傳球。梯度將在w.grad中提供。將訓練碼與循環中的體重更新一起運行，因爲單次更新可能不足以收斂。