美文网首页
pytorch 计算图与动态图机制

pytorch 计算图与动态图机制

作者: 米线织毛衣 | 来源:发表于2020-02-14 10:17 被阅读0次

    一、计算图

    (y对w的导数,其实就是y到w的所有路径上的导数之和)

    叶子节点是用户创建的节点,设置叶子节点可以节省内存,因为更新时,非叶子节点的梯度是会被释放的。

    二、动态图

    pytorch用的是动态图,tensorflow用的是静态图。

    静态图(先搭建好完整图,然后数据再流动):

    动态图(根据每一步搭建图):

    相关文章

      网友评论

          本文标题:pytorch 计算图与动态图机制

          本文链接:https://www.haomeiwen.com/subject/fqhdfhtx.html