D2L-6-计算图

Last updated Feb 2, 2022 Edit Source

• All
• Machine learning
• Deep learning

Tags: #MachineLearning #DeepLearning

• 将计算表示为一个无环图

例子 线性回归:

• 计算图有两种构造方法:
  • 显式构造
    • 主要应用于: Tensorflow/Theano/MXNet
    • 例子:

  1 2 3 4 5

  from mxnet import sym a = sym.var() b = sym.var() c = 2 * a + b # bind data into a and b later

  1 2 3 4 5 6 7 8 9

  **隐式构造** 主要应用于: `PyTorch/MXNet` 例子: `python om mxnet import autograd, nd th autograd.record(): = nd.ones((2,1))#创建两个二维向量 = nd.ones((2,1)) = 2 * a + b

  1