PyTorch的自动求导

2023年2月2日09:56:18

  PyTorch在autograd模块中实现了计算图的相关功能,autograd中的核心数据结构是Variable。Variable封装 了tensor,并记录对tensor的操作记录用来构建计算图。其中grad_fn得到了上一步相连的Tensor对应的grad(但如果设置requires_grad=False,最终就会将对应节点的grad置为None)。

  • 作者:herosunly
  • 原文链接:https://herosunly.blog.csdn.net/article/details/89535413
    更新时间:2023年2月2日09:56:18 ,共 179 字。