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为方差的标准正态分布中抽样。
Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐