Pytorch基礎函數（一）生成隨機張量——rand()與randn()

在PyTorch中，基本的數據單元爲Tensor（張量），而對於Tensor的隨機初始化，常用的就是rand()與randn()兩個函數。

一、rand()函數

torch.rand(*size,out=None)

輸出一個形狀爲size的均勻分佈Tensor。

例如：

print(torch.rand(2,3))

輸出：(兩行三列的均勻分佈Tensor)

二、randn()函數

torch..randn(*size,out=None)

輸出一個形狀爲size的標準正態分佈Tensor

例如：

print(torch.randn(3,2))

輸出：(三行兩列的正態分佈Tensor)

 

---end---

所以，rand()與randn()兩個均是用於生成隨機Tensor的函數，兩者的區別是rand生成的Tensor是基於均勻分佈的，而randn生成的向量是基於標準正態分佈的。