torch.randn和torch.rand函數


1.均勻分布
torch.rand(*sizes, out=None) → Tensor

返回一個張量,包含了從區間[0, 1)的均勻分布中抽取的一組隨機數。張量的形狀由參數sizes定義。

參數:  sizes (int...) - 整數序列,定義了輸出張量的形狀; out (Tensor, optinal) - 結果張量

例子:torch.rand(2, 3)

0.0836 0.6151 0.6958
0.6998 0.2560 0.0139
[torch.FloatTensor of size 2x3]

2.標准正態分布

當size是4個數時,返回一個2行3列的張量,其中每個元素又是一個4行2列的張量,最小元素的每一行服從正態分布


torch.randn(*sizes, out=None) → Tensor

返回一個張量,包含了從標准正態分布(均值為0,方差為1,即高斯白噪聲)中抽取的一組隨機數。張量的形狀由參數sizes定義。

參數:  sizes (int...) - 整數序列,定義了輸出張量的形狀;  out (Tensor, optinal) - 結果張量

例子:torch.randn(2, 3)

0.5419 0.1594 -0.0413
-2.7937 0.9534 0.4561
[torch.FloatTensor of size 2x3]

 

其他:離散正態分布

torch.normal(means, std, out=None) → → Tensor


原文鏈接:https://blog.csdn.net/wangwangstone/article/details/89815661


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM