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)函数作用:返回了一个张
·
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为方差的标准正态分布中抽样。
更多推荐
已为社区贡献1条内容
所有评论(0)