torch.randn和torch.rand有什麼區別
阿新 • • 發佈:2021-11-08
目錄
y = torch.rand(5,3) y=torch.randn(5,3)
y = torch.rand(5,3) y=torch.randn(5,3)
一個均勻分佈,一個是標準正態分佈。
torch.randn(2, 3)
torch.rand(2, 3)
均勻分佈
torch.rand(*sizes, out=None) → Tensor
返回一個張量,包含了從區間[0, 1)的均勻分佈中抽取的一組隨機數。
張量的形狀由引數sizes定義。
引數:
- sizes (int...) - 整數序列,定義了輸出張量的形狀
- out (Tensor, optinal) - 結果張量
標準正態分佈
**torch.randn(*sizes, out=None) ** → Tensor
返回一個張量,包含了從標準正態分佈(均值為0,方差為1,即高斯白噪聲)中抽取的一組隨機數。
張量的形狀由引數sizes定義。
引數:
- sizes (int...) - 整數序列,定義了輸出張量的形狀
- out (Tensor, optinal) - 結果張量
其他分佈取數:
離散正態分佈
torch.normal(means, std, out=None) → Tensor
線性間距向量
torch.linspace(start, end, steps=100, out=None) → Tensor
返回一個1維張量,包含在區間start和end上均勻間隔的step個點。
輸出張量的長度由steps決定。