PROSAGA码农传奇-深度学习-如何获取各种功能的mxnet后端代码？

<div class =“post-text”itemprop =“text”>
  <P>
    这实际上取决于符号图的样子。我假设你使用MXNet和Python（
    <a href="https://mxnet.incubator.apache.org/api/python/index.html" rel="nofollow noreferrer">
      Python文档
    </A>
    ）。在那里，您可以选择使用MXNet符号库或Gluon库。
  </p>
  <P>
    现在，你问是否可以检查代码，是的，你可以
    <a href="https://github.com/apache/incubator-mxnet" rel="nofollow noreferrer">
      在GitHub上找到它
    </A>
    。文件夹
     <code>
      python
    </code>
     包含python接口和
     <code>
      src
    </code>
     包含所有MXNet源。会发生什么事
     <code>
      forward
    </code>
     最终由MXNet执行引擎定义，该引擎跟踪运算符和神经网络层的输入/输出依赖性，在不同设备（CPU，GPU）上分配内存。
    <a href="http://mxnet.incubator.apache.org/versions/master/architecture/overview.html" rel="nofollow noreferrer">
      有一个通用的架构文档
    </A>
    。
  </p>
  <P>
    我想你对每个操作的作用感兴趣，比如
     <code>
      argmax
    </code>
     （减少），
     <code>
      tanh
    </code>
     （一元数学运算）或
     <code>
      convolution
    </code>
     （复杂的神经网络操作）。你可以在这里找到
    <a href="https://github.com/apache/incubator-mxnet/tree/master/src/operator" rel="nofollow noreferrer">
      MXNet的operator文件夹
    </A>
    。这需要一个完整的文档本身，并有一个MXNet细节的特殊论坛
    <a href="https://discuss.mxnet.io" rel="nofollow noreferrer">
      这里
    </A>
    ，但我会给出一个简短的指导：
  </p>
  <UL>
    <LI>
      （符号）执行图中的每个操作都需要定义
       <code>
        forward
      </code>
       和
       <code>
        backward
      </code>
       操作。它还需要定义其输出形状，以便可以与其他操作链接。如果该运算符需要权重，则需要定义所需的内存量，因此MXNet可以分配它。
    </LI>
    <LI>
      每个操作都需要几个实现a）CPU b）GPU（CUDA）c）cuDNN周围的包装
    </LI>
    <LI>
      所有一元数学运算都遵循相同的模式，因此它们都以类似的方式定义
      <a href="https://github.com/apache/incubator-mxnet/blob/master/src/operator/mshadow_op.h#L130" rel="nofollow noreferrer">
        mshadow_op.h
      </A>
       （例如。
       <code>
        relu
      </code>
      ）。
    </LI>
  </UL>
  <P>
    基于你这个相当广泛的问题，我可以告诉你这一切。
  </p>
</DIV>