久久国产成人av_抖音国产毛片_a片网站免费观看_A片无码播放手机在线观看,色五月在线观看,亚洲精品m在线观看,女人自慰的免费网址,悠悠在线观看精品视频,一级日本片免费的,亚洲精品久,国产精品成人久久久久久久

分享

PyTorch(二)——張量基本概念與常用操作

 taotao_2016 2019-09-10

張量(Tensor)

幾何代數(shù)中定義的張量是基于向量和矩陣的推廣,,通俗一點來理解,,我們可以將標量視為零階張量,矢量視為一階張量,,那么矩陣就是二階張量,。

PyTorch(二)——張量基本概念與常用操作

張量

在PyTorch中,,張量Tensor是最基礎的運算單位,,與NumPy中的NDArray類似,張量表示的是一個多維矩陣,。不同的是,,PyTorch中的Tensor可以運行在GPU上,而NumPy的NDArray只能運行在CPU上,。由于Tensor能在GPU上運行,,因此大大加快了運算速度。

可以通過rand( )函數(shù)生成一個簡單的張量,,例如生成一個2行3列0-1的隨機數(shù)Tensor,。

torch.rand(2, 3)

通過shape屬性或使用size( )函數(shù)可以查看Tensor的大小。

PyTorch(二)——張量基本概念與常用操作

查看Tensor大小

在同構的意義下,,我們設r為張量的秩或階,,那么,第零階張量(r = 0)為標量,,第一階張量(r = 1)為向量,,第二階張量(r = 2)為矩陣,第三階以上(r > 2)的統(tǒng)稱為多維張量,。

Tensor基本數(shù)據(jù)類型

Tensor的常用基本數(shù)據(jù)類型主要有以下五種:

  1. 32位浮點型:torch.FloatTensor,。Tensor的默認數(shù)據(jù)類型。
  2. 64位浮點型:torch.DoubleTensor,。
  3. 64位整型:torch.LongTensor,。
  4. 32位整型:torch.IntTensor。
  5. 16位整:torch.ShortTensor,。

此外,,Tensor的數(shù)據(jù)類型還可以是byte或chart型。

PyTorch(二)——張量基本概念與常用操作

Tensor基本數(shù)據(jù)類型

Tensor初始化

除了使用rand( )函數(shù)外,,PyTorch中還提供了許多初始化張量的方法,,可以類比NumPy中對NDArray的初始化。

randn( ):初始化一個均值為0,,方差為1的隨機數(shù)Tensor,。

ones( ):初始化一個全為1的Tensor。

zeros( ):初始化一個全為0的Tensor,。

eye( ):初始化一個主對角線為1,,其余都為0的Tensor(只能是二階,,即矩陣)。

PyTorch(二)——張量基本概念與常用操作

Tensor初始化

Tensor常用API

PyTorch對Tensor的操作與NumPy對NDArray的操作非常相似,,因此在學習Tensor時可以經(jīng)常對比NDArray,。事實上,它們兩者基本就是一致的,。

max( )/min( ):沿行或列取最大值/最小值,。參數(shù)dim=0表示沿列,dim=1表示沿行,。返回value(值)和idx(索引),。

PyTorch(二)——張量基本概念與常用操作

max()/min()

sum( ):沿行/列求和。

PyTorch(二)——張量基本概念與常用操作

sum()

基本四則運算,。

PyTorch(二)——張量基本概念與常用操作

基本四則運算

add( )與add_( ):以_為結尾的會改變調用值本身,。

PyTorch(二)——張量基本概念與常用操作

add( )與add_( )

mm( ):矩陣乘法。

PyTorch(二)——張量基本概念與常用操作

mm( )

PyTorch的Tensor與NumPy的NDArray轉換

使用PyTorch的numpy( )函數(shù)將Tensor轉換為NDArray,。

PyTorch(二)——張量基本概念與常用操作

Tensor轉NDArray

使用PyTorch的from_numpy( )函數(shù)將NDArray轉換為Tensor,。

PyTorch(二)——張量基本概念與常用操作

NDArray轉Tensor

Tensor運行設備(GPU與CPU)轉換

一般情況下可以使用cuda( )函數(shù)將Tensor轉移到GPU上運行,當然這步操作需要cuda設備的支持,。

PyTorch(二)——張量基本概念與常用操作

將Tensor轉移到GPU上運行

    本站是提供個人知識管理的網(wǎng)絡存儲空間,,所有內容均由用戶發(fā)布,不代表本站觀點,。請注意甄別內容中的聯(lián)系方式,、誘導購買等信息,謹防詐騙,。如發(fā)現(xiàn)有害或侵權內容,,請點擊一鍵舉報。
    轉藏 分享 獻花(0

    0條評論

    發(fā)表

    請遵守用戶 評論公約

    類似文章 更多