pytorch学习笔记1

  • torch.randn_like()
    function:生成新的张量,保持原始张量的size及数据类型
a = torch.ones(3,3, dtype = torch.float)
x = torch.randn_like(a)    # x 的size为3×3,且dtype为torch.float
  • tensor_1.data.norm(p)
    function: 得到张量tensor_1的p范数,默认为2范数
a = torch.rand(3)
y = a.data.norm(3)  #  得到 a 的3范数
# 或者 y = torch.norm(a,3)   
  • out.backward()
    function:进行反向传播,计算梯度
x = torch.randn(3, requires_grad = True)
y = (x*3).sum()
y.backward()   
print(x.grad)  

注意! 此时的out必须是标量,否则无法计算

在不断地进行1.更新y和2.y.backward() 两个步骤时 ,梯度会累加(这里的y仅仅代表张量名,每次都可以不一样)。若重新计算梯度,则可调用optimizer.zero_grad()“清空”梯度。
梯度累加

  • y.backward(v)
    function: 根据链式法则,通过计算vector与Jacobian matrix的乘积得到梯度 ∂ l / ∂ x ⃗ \partial l/\partial \vec x l/x (如下图)
    y , v , l y,v,l y,v,l的关系: l l l关于 y y y的函数为 l = g ( y ⃗ ) l=g(\vec{y}) l=g(y ) v = ( ∂ l ∂ y 1 . . . ∂ l ∂ y m ) T v={(\frac {\partial l}{\partial y_1} ... \frac {\partial l}{\partial y_m} )}^T v=(y1l...yml)T
    雅克比矩阵
    vector-Jacobian product
x = torch.randn(3, requires_grad = True)
y = x*3

v = torch.tensor([1.0, 0.5, 1.0], dtype=torch.float)
y.backward(v)
print(x.grad)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值