Как код 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

Другие вопросы по тегам