gpt4 book ai didi

python - MXNET C++ 代码如何与其 python API 绑定(bind)?

转载 作者:太空宇宙 更新时间:2023-11-03 14:32:21 24 4
gpt4 key购买 nike

1.使用C++在MXNet中定义自定义运算符时,C++中定义的自定义运算符如何自动生成其Python API?2.如何找到对应的代码?3. 使用C++和Python定义自定义运算符有什么区别?

最佳答案

宏 NNVM_REGISTER_OP 按名称注册一个运算符,然后从 python 中可以看到该运算符。从那里,您可以定义运算符的属性(即 FCompute,它是调用时要调用的函数)。

在此处的 C++ 部分,有一个指南:https://mxnet.incubator.apache.org/how_to/new_op.html

关于python - MXNET C++ 代码如何与其 python API 绑定(bind)?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47193488/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com