Как код MXNET C++ связывается с его Python API?
1.При определении пользовательского оператора в MXNet с C++, как пользовательский оператор, определенный в C++, автоматически генерирует свой Python API? 2.Как я могу найти соответствующие коды? 3. В чем разница между определением пользовательского оператора в C++ и в python?
1 ответ
Макрос NNVM_REGISTER_OP регистрирует оператор по имени, которое затем видно из python. Оттуда вы определяете атрибуты оператора (то есть FCompute, который является функцией, вызываемой при вызове).
В разделе C++ здесь есть руководство: https://mxnet.incubator.apache.org/how_to/new_op.html