onunix 2023-02-11 14:08 采纳率: 78.9%
浏览 98
已结题

请教一下torch.autograd.grad(y,x,create_graph=True)[0]这里的[0]是什么意思?

请教一下pytorch的这段代码里:torch.autograd.grad(y,x,create_graph=True)[0],这里

import numpy as np 
import torch 

# f(x) = a*x**2 + b*x + c的导数

x = torch.tensor(0.0,requires_grad = True) # x需要被求导
a = torch.tensor(1.0)
b = torch.tensor(-2.0)
c = torch.tensor(1.0)
y = a*torch.pow(x,2) + b*x + c


# create_graph 设置为 True 将允许创建更高阶的导数 
dy_dx = torch.autograd.grad(y,x,create_graph=True)[0]
print(dy_dx.data)

# 求二阶导数
dy2_dx2 = torch.autograd.grad(dy_dx,x)[0] 

print(dy2_dx2.data)

请问 torch.autograd.grad(y,x,create_graph=True)[0],这里为什么有[0] ? 既然可以[0],那么 torch.autograd.grad(y,x,create_graph=True)[1]又表示什么呢?

  • 写回答

5条回答 默认 最新

  • 元气少女缘结神 2023-02-11 15:20
    关注

    的确,都说torch.autograd.grad返回的是第一参数对第二个参数的导数,一般是网络中的梯度,为什么都只取[0],我也要去看下笔记查下。这一维度的具体意义

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(4条)
编辑
预览

报告相同问题?

问题事件

  • 系统已结题 2月19日
  • 已采纳回答 2月12日
  • 创建了问题 2月11日

悬赏问题

  • ¥15 PADS Logic 原理图
  • ¥15 PADS Logic 图标
  • ¥15 电脑和power bi环境都是英文如何将日期层次结构转换成英文
  • ¥20 气象站点数据求取中~
  • ¥15 如何获取APP内弹出的网址链接
  • ¥15 wifi 图标不见了 不知道怎么办 上不了网 变成小地球了
手机看
程序员都在用的中文IT技术交流社区

程序员都在用的中文IT技术交流社区

专业的中文 IT 技术社区,与千万技术人共成长

专业的中文 IT 技术社区,与千万技术人共成长

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

客服 返回
顶部