使用PyTorch实现自动求导

本文介绍了如何使用PyTorch进行自动求导,通过四个具体的例子展示了不同操作的梯度计算,包括:y = 2 * dot(x, x),x.sum(),y = x * x以及y = u * x(u为标量)。每个例子都验证了导数的正确性,强调在深度学习中,我们关注的是批量样本的偏导数之和。" 82260091,5808315,深入理解Spring注解@Import与@ImportResource,"['JAVA', 'Spring', 'Spring Boot', '配置管理']
摘要由CSDN通过智能技术生成
  1. 准备工作
import torch

x = torch.arange(4.0)

# 在计算y关于x的梯度前,需要一个地方来储存梯度
x.requires_grad_(True)

也可以直接在创建tensor的时候直接定义requires_grad

torch.arange(4.0, requires_grad=True)
<
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值