mxnet系列之 代码阅读1
在文献[1]中写了怎样读mxnet的源代码。
所有的模块接口的定义都在include/mxnet中,意思就是这个include里面的代码就是好的document
以base.h为例进行分析
#ifndef MXNET_BASE_H_ #define MXNET_BASE_H_ #include <dmlc/base.h> #include <dmlc/io.h> #include <dmlc/type_traits.h> #include <dmlc/parameter.h> #include <mshadow/tensor.h> // nnvm headers for symbolic construction. #include <nnvm/op.h> #include <nnvm/tuple.h> #include <nnvm/symbolic.h> #include <string>可以看出,使用了dmlc、mshadow、nnvm三个框架里面的函数
[1] read mxnet code