跳转至

自动求导

基础

  • 创建可自动求导张量的方法:
  • 对张量调用requires_grad_(True)方法
  • 创建张量时指定参数requires_grad=True
  • 对张量的函数求导:调用backward()方法
  • 张量清零:对张量的grad对象调用zero_()方法,例:x.grad.zero_()
  • detach()方法:将张量剥离计算图,即当作不可求导参数使用