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 

posted @ 2017-04-21 15:42  开往春天的拖拉机  阅读(138)  评论(0编辑  收藏  举报