Pytorch中的自動求梯度機制和Variable類實例
自動求導(dǎo)機制是每一個深度學(xué)習(xí)框架中重要的性質(zhì),免去了手動計算導(dǎo)數(shù),下面用代碼介紹并舉例說明Pytorch的自動求導(dǎo)機制。
首先介紹Variable,Variable是對Tensor的一個封裝,操作和Tensor是一樣的,但是每個Variable都有三個屬性:Varibale的Tensor本身的.data,對應(yīng)Tensor的梯度.grad,以及這個Variable是通過什么方式得到的.grad_fn,根據(jù)最新消息,在pytorch0.4更新后,torch和torch.autograd.Variable現(xiàn)在是同一類。torch.Tensor能像Variable那樣追蹤歷史和反向傳播。Variable仍能正確工作,但是返回的是Tensor。
我們擁抱這些新特性,看看Pytorch怎么進行自動求梯度。
#encoding:utf-8 import torch x = torch.tensor([2.],requires_grad=True) #新建一個tensor,允許自動求梯度,這一項默認是false. y = (x+2)**2 + 3 #y的表達式中包含x,因此y能進行自動求梯度 y.backward() print(x.grad)
輸出結(jié)果是:
tensor([8.])
這里添加一個小知識點,即torch.Tensor和torch.tensor的不同。二者均可以生成新的張量,但torch.Tensor()是python類,是默認張量類型torch.FloatTensor()的別名,使用torch.Tensor()會調(diào)用構(gòu)造函數(shù),生成單精度浮點類型的張量。
而torch.tensor()是函數(shù),其中data可以是list,tuple,numpy,ndarray,scalar和其他類型,但只有浮點類型的張量能夠自動求梯度。
torch.tensor(data, dtype=None, device=None, requires_grad=False)
言歸正傳,上一個例子的變量本質(zhì)上是標量。下面一個例子對矩陣求導(dǎo)。
#encoding:utf-8 import torch x = torch.ones((2,4),requires_grad=True) y = torch.ones((2,1),requires_grad=True) W = torch.ones((4,1),requires_grad=True) J = torch.sum(y - torch.matmul(x,W)) #torch.matmul()表示對矩陣作乘法 J.backward() print(x.grad) print(y.grad) print(W.grad)
輸出結(jié)果是:
tensor([[-1., -1., -1., -1.], [-1., -1., -1., -1.]]) tensor([[1.], [1.]]) tensor([[-2.], [-2.], [-2.], [-2.]])
以上這篇Pytorch中的自動求梯度機制和Variable類實例就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
Django + Uwsgi + Nginx 實現(xiàn)生產(chǎn)環(huán)境部署的方法
Django的部署可以有很多方式,采用nginx+uwsgi的方式是其中比較常見的一種方式。這篇文章主要介紹了Django + Uwsgi + Nginx 實現(xiàn)生產(chǎn)環(huán)境部署,感興趣的小伙伴們可以參考一下2018-06-06
Django 后臺獲取文件列表 InMemoryUploadedFile的例子
今天小編就為大家分享一篇Django 后臺獲取文件列表 InMemoryUploadedFile的例子,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
Python深度學(xué)習(xí)之使用Pytorch搭建ShuffleNetv2
這篇文章主要介紹了Python深度學(xué)習(xí)之使用Pytorch搭建ShuffleNetv2,文中有非常詳細的圖文示例,對正在學(xué)習(xí)python的小伙伴們有非常好的幫助,需要的朋友可以參考下2021-05-05
Python中zip()函數(shù)的解釋和可視化(實例詳解)
zip() 函數(shù)用于將可迭代的對象作為參數(shù),將對象中對應(yīng)的元素打包成一個個元組,然后返回由這些元組組成的列表。這篇文章主要介紹了Python中zip()函數(shù)的解釋和可視化,需要的朋友可以參考下2020-02-02
django應(yīng)用JWT(JSON?Web?Token)實戰(zhàn)教程
在前后端分離的項目中,JWT(JSON?Web?Token)作為一種廣泛使用的身份驗證和授權(quán)機制,提供了一種安全、高效的方式來保護RESTful?API,本文詳細介紹了JWT的概念、優(yōu)勢、在Django中的應(yīng)用步驟和使用方法,是構(gòu)建安全、高效Web應(yīng)用的有效指南2024-10-10

