Как написать пользовательский уровень MXNet с узнанными параметрами

Question

Я следую документации в http://mxnet.io/how_to/new_op.html, чтобы определить новый уровень нейронной сети в MXNet в python путем подкласса класса mx.operator.CustomOp. Примером является слой потерь, который не имеет выученных параметров. Итак, как же полученные параметры попадают в методы forward и backward?

Answer 1

Я понял это. Выбранные параметры сконфигурированы как любой другой вход в op. Они настроены в методе list_arguments. Из docs page on writing custom symbols:

Обратите внимание, что список аргументов объявляет ввод и параметр, и мы рекомендуем заказывать их ['input1', 'input2', ... , 'weight1', 'weight2', ...]