【发布时间】:2017-11-09 03:35:56
【问题描述】:
1.在MXNet中用C++定义自定义算子时,C++中定义的自定义算子是如何自动生成其python API的? 2.如何找到对应的代码? 3. 用C++和python定义自定义操作符有什么区别?
【问题讨论】:
标签: python deep-learning mxnet custom-operator
1.在MXNet中用C++定义自定义算子时,C++中定义的自定义算子是如何自动生成其python API的? 2.如何找到对应的代码? 3. 用C++和python定义自定义操作符有什么区别?
【问题讨论】:
标签: python deep-learning mxnet custom-operator
宏 NNVM_REGISTER_OP 按名称注册一个运算符,然后在 python 中可见。从那里,您可以定义运算符的属性(即 FCompute,它是调用时要调用的函数)。
这里的C++部分有一个指南:https://mxnet.incubator.apache.org/how_to/new_op.html
【讨论】: