RicardoLu
  • Joined on Mar 21, 2023
Loading Heatmap…

RicardoLu commented on issue Datawhale/d2l#23

关于计算图的问题

我的理解,计算图是有向无环图,但是图中的一个节点可以有很多条边,所以代码中的x通过多条边指向多个节点,整个图中只要无环即可。 你的截图中的代码的计算图是下面这样的: ![image](/attachments/a9faac02-d864-48e4-9680-ea9b6a81536e) 青色矩形为叶子节点x,会绑定一个AccumulateGrad操作累积梯度。可以看到叶子结点通过多条边和其他节点相连。 假如修改代码中的y = x * x * x,那么计算图会变成下面这样: ![image](/attachments/9423f782-033c-4a68-ba9f-d6e47d09f9ab) 在计算y时产生了一个临时节点存储 x * x。 如果接着修改z = x + y + x,那么计算图会变成这样: ![image](/attachments/153fa1f5-c29c-4ff7-8281-47bb22191898) 我觉得图大概挺直观了。

1 year ago