文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Pytorch常用的4种随机数生成方法

2023-09-21 16:41

关注

Pytorch常用的4种随机数生成方法

一、torch.rand():构造均匀分布张量的方法

torch.rand是用于生成均匀随机分布张量的函数,从区间[0,1)的均匀分布中随机抽取一个随机数生成一个张量,其调用方法如下所示:

torch.rand(sizes, out=None) ➡️ Tensor

其中,

import torch# 生成一个每个元素服从0-1均匀分布的4行3列随机张量tensor_1 = torch.rand(4, 3)print(tensor_1, tensor_1.type())

输出结果如下图所示:
在这里插入图片描述

二、torch.randn():构造标准正态分布张量的方法

torch.randn()是用于生成正态随机分布张量的函数,从标准正态分布中随机抽取一个随机数生成一个张量,其调用方法如下所示:

torch.randn(sizes, out=None) ➡️ Tensor

其中,

import torch# 生成一个每个元素均为标准正态分布的4行3列随机张量tensor_2 = torch.randn(4, 3)print(tensor_2, tensor_1.type())

输出结果如下图所示:
在这里插入图片描述

三、torch.randint():构造区间分布张量的方法

torch.randint()是用于生成任意区间分布张量的函数,从标准正态分布中随机抽取一个随机数生成一个张量,其调用方法如下所示:

torch.randint(low=0, high, sizes, out=None) ➡️ Tensor

其中,

# 生成一个每个元素均为[1-10]均匀分布的4行3列随机张量tensor_3 = torch.randint(1, 10, (4, 3))print(tensor_3, tensor_3.type())

输出结果如下图所示:
在这里插入图片描述

四、torch.randperm():根据生成的随机序号对张量进行随机排序的方法

torch.randint()是用于对张量序号进行随机排序的函数,并根据生成的随机序列,其调用格式如下所示:

torch.randperm(n, out=None, dtype=torch.int64) ➡️ LongTensor

其中,

简单的示例代码如下所示:

下面代码实现的功能为:将二维张量按照行进行随机排序。

# 生成一个0~3的随机整数排序idx = torch.randperm(4)# 生成一个4行3列的张量tensor_4 = torch.Tensor(4, 3)# 为了方便对比,首先输出tensor_4的结果print("原始张量\n", tensor_4)# 下面输出随机生成的行序号print("\n生成的随机序号\n", idx)# 下面的指令实现了在行的方向上,对tensor_4进行随机排序,并输出结果print("\n随机排序后的张量\n", tensor_4[idx])

输出结果如下图所示:
在这里插入图片描述

来源地址:https://blog.csdn.net/weixin_37926734/article/details/126614099

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯