摘要: 一、自动求导系统 torch.autograd 1、torch.autograd.backward 功能:自动求取梯度 tensors:用于求导的张量,如loss retain_graph:保存计算图 create_graph:创建导数计算图,用于高阶求导 grad_tensors:多梯度权重 te 阅读全文
posted @ 2019-10-17 21:01 cola_cola 阅读(220) 评论(0) 推荐(0) 编辑