mxnet - 如何在 mxnet 中从 python 跳转到 c++

Question

现在我正在阅读 mxnet 中的源代码。读取数据的代码是train_dataiter = mx.io.MNISTIter()。我发现类MNISTIter: public IIterator {}是在src/io中实现的。怎么能从python跳转到c++？一直想知道怎么包装的。谁能帮忙解释一下？

Answer 1

在 cpp 中实现的数据迭代器在运行时注册为 python 函数。这里有一些相关的功能可以做到这一点：https ://github.com/dmlc/mxnet/blob/master/python/mxnet/io.py#L778