【PyTorch学习】 生成随机数

torch.rand

返回一个张量,包含了从区间[0, 1)的均匀分布中抽取的一组随机数。张量的形状由参数sizes定义。


tmp = torch.rand(2, 3)


torch.randn

返回一个张量,包含了从标准正态分布(均值为0,方差为1,即高斯白噪声)中抽取的一组随机数。张量的形状由参数sizes定义。

tmp = torch.randn(2, 3)

 


torch.normal

返回一个张量,包含了从指定均值means和标准差std的离散正态分布中抽取的一组随机数。

标准差std是一个张量,包含每个输出元素相关的正态分布标准差。

print(torch.normal(mean=0.5, std=torch.arange(1, 6).float()))

 


torch.linspace

返回一个1维张量,包含在区间start和end上均匀间隔的step个点。

输出张量的长度由steps决定。

print(torch.linspace(1, 10, steps=4))

 

 

 

发布了83 篇原创文章 · 获赞 14 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/weixin_38121168/article/details/103097474