PyTorch生成随机数的各种常见方法

在使用PyTorch做实验时经常会用到生成随机数Tensor的方法,比如:

torch.rand()

torch.randn()

torch.normal()

torch.linespace()

在很长一段时间里我都没有区分这些方法生成的随机数究竟有什么不同,由此在做实验的时候经常会引起一些莫名其妙的麻烦。

所以在此做一个总结,以供大家阅读区分,不要重蹈我的覆辙。

均匀分布

torch.rand(sizes, out=None)* → Tensor

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

参数:

  • sizes (int…) - 整数序列,定义了输出张量的形状
  • out (Tensor, optinal) - 结果张量

例子:

torch.rand(2, 3)
0.0836 0.6151 0.69580.6998 0.2560 0.0139[torch.FloatTensor of size 2x3]

标准正态分布

torch.randn(sizes, out=None)* → Tensor

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

参数:

  • sizes (int…) - 整数序列,定义了输出张量的形状
  • out (Tensor, optinal) - 结果张量

例子:

torch.randn(2, 3)
0.5419 0.1594 -0.0413-2.7937 0.9534 0.4561[torch.FloatTensor of size 2x3]

离散正态分布

torch.normal(means, std, out=None) → → Tensor

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

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

参数:

  • means (float, optional) - 均值
  • std (Tensor) - 标准差
  • out (Tensor) - 输出张量

例子:

torch.normal(mean=0.5, std=torch.arange(1, 6))
-0.1505-1.2949-4.4880-0.5697-0.8996[torch.FloatTensor of size 5

线性间距向量

torch.linspace(start, end, steps=100, out=None) → Tensor

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

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

参数:

  • start (float) - 区间的起始点
  • end (float) - 区间的终点
  • steps (int) - 在start和end间生成的样本数
  • out (Tensor, optional) - 结果张量

例子:

torch.linspace(3, 10, steps=5)
3.00004.75006.50008.250010.0000[torch.FloatTensor of size 5]

原文地址:

集智学园的文字

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 后文主要是对pytorch官网的文档的总结。代码来自pytorch中文API torch.squeeze(inpu...
    永远学习中阅读 12,517评论 0 1
  • 本篇内容参考官方文档自己总结而来仅供自学自查,详细需求请查阅官方文档。 数据类型 张量(Tnsor) 什么是张量?...
    Zimix阅读 7,733评论 0 3
  • 最近一段时间感于自己的写作能力羸弱,于是转头向各位文学大师学习,断断续续的看了一个月左右的古典文学,比如狄更斯的《...
    李子心诚阅读 1,225评论 5 4
  • 这里列出的5款linux发行版,它们包具备很好的的渗透测试工具以及其他黑客工具.这里列表中目前最受欢迎的Linux...
    等会再说阅读 10,687评论 10 101
  • 今天雨很大,在售楼处等到中午我就回来了,乙方仍然不来人。正准备出去买饭,一朋友打电话说来衡水看看我,碰面后她说她...
    艳云阅读 269评论 0 0

友情链接更多精彩内容