在文献[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
所有评论(0)