1.在MXNet中用C++定义自定义算子时，C++中定义的自定义算子是如何自动生成其python API的？2.如何找到对应的代码？3. 用C++和python定义自定义操作符有什么区别？
问问题
263 次
1 回答
2
宏 NNVM_REGISTER_OP 按名称注册一个运算符，然后从 python 中可见。从那里，您可以定义运算符的属性（即 FCompute，它是调用时要调用的函数）。
在这里的 C++ 部分，有一个指南： https ://mxnet.incubator.apache.org/how_to/new_op.html
于 2017-11-20T21:49:40.957 回答