Pytorch的API詳解

ygpGoogle發表於2020-12-05

匯入torch庫,然後用torch.empty(5,3)初始化一個5*3的tensor。
這個tensor裡的數字是隨機的。
在這裡插入圖片描述
torch.rand(5,3)裡建立的tensor裡的數字是零到一的數字。
在這裡插入圖片描述
建立一個全是0的tensor,型別預設是torch.float32
在這裡插入圖片描述
還可以自己制定型別
在這裡插入圖片描述
還可以使用long()函式
在這裡插入圖片描述
還可以根據已有資料建立tensor。
在這裡插入圖片描述
也可以從一個已有的tensor建立新的tensor,新建的tensor會有之前的tensor的一些特徵,如下圖:我用new_ones(5,3)建立了一個全是1的tensor,dtype型別是舊的x中數的型別,當然,我也可以手動指定型別,如torch.double
在這裡插入圖片描述
利用randn_like函式可以產生與x形狀相同的隨機tensor
在這裡插入圖片描述

shape屬性表示形狀

在這裡插入圖片描述
tensor的加法
在這裡插入圖片描述
在這裡插入圖片描述
還有一種in-place形式的加法:
像這種有下劃線的方法,都會改變呼叫者原有的值。
在這裡插入圖片描述
在這裡插入圖片描述
各種類似numpy的索引切片操作都可以在tensor上搞。
在這裡插入圖片描述
這裡的view就相當於numpy的reshape
在這裡插入圖片描述
在這裡插入圖片描述
未完待續

相關文章