MXNET C++ 代码如何与其 python API 绑定？

Posted 2023-02-23

【中文标题】MXNET C++ 代码如何与其 python API 绑定？

【英文标题】：How does MXNET C++ code bind with its python API?

【发布时间】：2017-11-09 03:35:56

【问题描述】：

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