pytorch tensor創建

create tensor

1. torch.tensor():從data創建tensor

data可以是list,numpy;dtype默認與data一致。

2. torch.from_numpy(ndarray):從numpy創建tensor

這裏創建的tensor與原ndarray共享內存,一個改變另一個就會改變。

3.

torch.zeros():依size創建全0張量

 

torch.zeros_like():依input形狀創建全0張量

4.

torch.ones()

torch.ones_like()

5.

torch.full():依size創建全fill_value張量

torch.full_like()

6. torch.arange():創建等差的一維張量,創建的區間爲[start, end)

7. torch.linspace():創建均分的一維張量,數值區間爲[start, end]

8. torch.eye():創建單位對角矩陣(二維張量),默認爲方陣

9. 

torch.normal():生成正態分佈的張量

mean和std可以爲標量和張量

當mean和std都爲標量時,要加一個參數size來指定tensor的大小

torch.randn()torch.randn_like():生成標準正態分佈的tensor

torch.rand()torch.ran_like():在區間[0, 1)上生成均勻分佈

torch.randint()torch.randint_like():區間[low, high]生成整數均勻分佈

 

torch.randperm():生成從0到n-1的隨機排列

torch.bernoulli():以input爲概率,生成伯努利分佈

 

 

 

 

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