如何使用學習參數編寫自定義MXNet層

Question

我正在按照http://mxnet.io/how_to/new_op.html中關於如何在python中通過繼承mx.operator.CustomOp類來定義MXNet中的新神經網絡層的文檔。該示例是沒有學習參數的丟失層。那麼學習參數如何進入forward和backward方法？

Answer 1

我想通了。學習到的參數像任何其他輸入一樣配置。它們的配置方法爲list_arguments。從docs page on writing custom symbols：

注意列表參數聲明輸入和參數，我們 建議，命令他們作爲['input1', 'input2', ... , 'weight1', 'weight2', ...]