torch.rand()与torch.randn()的用法与区别

torch.rand*sizes, out=None)

函数作用：

返回了一个张量，
包含了从0-1的均匀分布中抽取的一组随机数，
张量的形状由size定义。

函数参数

sizes：生成张量的形状。

函数举例

y=torch.rand(2, 2, 3)
print(y)

输出：

生成了两个二行三列的数组，且取值从0，1之间的均匀分布中抽样。

torch.randn(*sizes, out=None)

函数作用：

返回了一个张量，
包含了从标准正态分布（均值为0，方差为1）中抽取的一组随机数。
张量的形状由size定义。

函数参数

sizes：生成张量的形状和大小

函数举例

x=torch.randn(2, 2, 3)
print(x)

输出：

同样是两个二行三列的数组，
取值是从以0为均值，1为方差的标准正态分布中抽样的。

总结：

rand（）
randn（）
相同：两个函数参数sizes都是定义返回张量的形状。

不同：

rand（）取值从0，1之间的均匀分布中抽样。
randn（）取值从以0为均值，1为方差的标准正态分布中抽样。