Pytorch入门---学习笔记四---Autograd与逻辑回归

Autograd—自动求导系统 torch.autograd.backward 功能:自动求取梯度 tensors:用于求导的张量,如loss retain_graph:保存计算图 create_graph:创建导数计算图,用于高阶求导 grad_tensors:多梯度权重
相关文章
相关标签/搜索