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