我正在关注http://mxnet.io/how_to/new_op.html中的文档，了解如何通过子类化类在 python 中的 MXNet 中定义新的神经网络层mx.operator.CustomOp。该示例是没有学习参数的损失层。那么学习到的参数是如何进入forward和backward方法的呢？
问问题
647 次
1 回答
3
我想通了。学习参数的配置与操作的任何其他输入一样。它们是在list_arguments方法中配置的。从关于编写自定义符号的文档页面：
请注意，列表参数声明了输入和参数，我们建议将它们排序为
['input1', 'input2', ... , 'weight1', 'weight2', ...]
于 2016-11-21T18:07:52.623 回答