pytorch初始化 創建tensor(均勻採樣、正態分佈採樣等)

均勻採樣

torch.rand()在[0,1]中均勻採樣
torch.rand_like()
*_like標籤的接收的是tensor
在這裏插入圖片描述

torch.randint(1,10,[3,3])隨機的是[1,10)的整數[3,3]爲shape
示例1

full的使用

在這裏插入圖片描述
torch.full([],7)創建的是標量
torch.full([1],7)創建的是tensor
在這裏插入圖片描述

正態分佈採樣

torch.randn(3,3)均值爲0方差爲1的[3,3]tensor
在這裏插入圖片描述
torch.full([10],0)長度爲10,值都爲0的tensor
torch.arange(1,0,-0.1)從1開始向0減少 一次減0.1
在這裏插入圖片描述
torch.normal正態分佈
在這裏插入圖片描述

arange/range

第三個參數是階梯
不推薦使用range函數
在這裏插入圖片描述

linspace/logspace

steps爲數量值
arange中爲階梯
torch.linspace(0,10,steps=11)
linspace爲[0,10]
torch.logspace(0,-1,steps=10)
logspace爲100~10-1

在這裏插入圖片描述

Ones/zeros/eye

輸入shape
ones爲全1
zeros爲全0
eye爲對角線,適合矩陣
在這裏插入圖片描述
在這裏插入圖片描述
在這裏插入圖片描述

randperm

隨機打散
在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章