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