在 forward 的每一步里,比如加法、乘法、Relu、等等,如何做 backward 就已经记录到了结果的 tensor 里。所以可以在最终结果 tensor 里直接调用 backward 。
阅读量:1
点赞量:0