【深度之眼】Pytorch框架班第五期-Week1任务3第一节:autograd与逻辑回归
torch.autograd
autograd-自动求导系统
torch.autograd.backward(tensors, grad_tensors=None, retain_graph=None, create_graph=False)
功能:自动求取梯度
tensors: 用于求导的张量, 如loss
retain_graph: 保存计算图
create_graph: 创建导数计算图,用于高阶求导
grad_tensors: 多梯度权重
retain_graph
import torch
imp
原创
2020-07-24 18:28:54 ·
273 阅读 ·
0 评论