PyTorch学习笔记之Variable

本文介绍了PyTorch中的Autograd模块,重点讲解了Variable类的功能和属性,包括data、grad和grad_fn。Variable用于自动求导,通过.backward()方法进行反向传播,计算梯度。还提到了requires_grad和volatile参数的作用,以及如何处理梯度积累问题。
摘要由CSDN通过智能技术生成

深度学习的算法本质上是通过反向传播求导数,PyTorch的Autograd模块实现了此功能。

autograd.Variable是Autograd中的核心类,它简单封装了Tensor,并支持几乎所有Tensor的操作。Tensor在被封装为Variable之后,可以调用它的.backward实现反向传播,自动计算所有梯度。

Variable主要包含三个属性

  • data: 保存了Variable所包含的Tensor
  • grad: 保存data对应的梯度,grad也是个Variable,而不是Tensor,它和data的形状一样
  • grad_fn: 指向一个Function对象,这个Function用来反向传播计算输入的梯度
from torch.autograd import Variable
# 使用Tensor新建一个Variable
x = Variable(t.ones(2, 2)<
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值