在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生成的向量是基于标准正态分布的。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐