mxnet系列之代码阅读1

最新推荐文章于 2024-09-21 22:34:33 发布

andeyeluguo

最新推荐文章于 2024-09-21 22:34:33 发布

阅读量1.6k

点赞数

CC 4.0 BY-SA版权

分类专栏： mxnet

本文链接：https://blog.youkuaiyun.com/andeyeluguo/article/details/70319008

mxnet 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了如何阅读MXNet源代码，重点分析了include/mxnet目录下的base.h文件。此文件包含了dmlc、mshadow及nnvm三个框架的相关头文件，揭示了MXNet核心模块的构建方式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在文献[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