pytorch初始化张量并填充随机整数值

devtools/2024/10/9 0:04:03/

在 PyTorch 中,你可以使用 torch.randint 或 torch.empty 加上 random_ 方法来初始化张量并填充随机整数值。以下是两种常用的方法:

1. 使用 torch.randint

torch.randint 直接生成一个指定形状的张量

,并填充在给定范围内的随机整数。

示例代码:
import torch# 生成一个 3x3 的张量,填充 0 到 9 之间的随机整数(不包括 10)
tensor = torch.randint(low=0, high=10, size=(3, 3))print(tensor)

2. 使用 torch.empty 和 random_

torch.empty 先生成一个未初始化的张量,然后用 random_ 方法填充随机整数值。

示例代码:
import torch# 生成一个未初始化的 3x3 张量
tensor = torch.empty(3, 3, dtype=torch.long)# 用 0 到 9 之间的随机整数(不包括 10)填充张量
tensor = tensor.random_(10)print(tensor)

解释:

  • torch.randint 是更直接的方式,一步完成张量的初始化和填充。
  • torch.empty + random_ 的组合提供了更多的控制权,比如可以指定张量的 dtype

这两种方法都可以用于生成指定形状和范围的随机整数张量。


http://www.ppmy.cn/devtools/103500.html

相关文章

JavaScript 中的 Stream API 02(可读流 ReadableStream 等 API)

Stream API 可读流 API ReadableStream ReadableStream 可读流实例创建: new ReadableStream(underlyingSource, queuingStrategy) underlyingSource 参数(可选): 该参数包含构造流对应的行为方法和属性的对象,主要包含如下 start(control…

vue3项目使用EventSource实现流式输出例如滚动日志

前言 之前接触的通信方式主要是HTTP请求和WebSocket,这次有机会了解到EventSource,记录一下。 简介 EventSource是一个浏览器端用于接收服务器推送事件(Server-Sent Events, SSE)的 JS API。与 WebSocket 不同,SSE …

Linux CentOS 添加路由

一、临时添加 查看当前路由表 ip route show添加路由 假设你要添加一条到特定网络的路由&#xff0c;可以使用以下命令&#xff1a; sudo ip route add <destination_network>/<netmask> via <gateway> dev <interface>例如&#xff0c;要添加一条…

Dockerfile应用、私有仓库

通过dockerfile创建⼀个在启动容器时&#xff0c;就可以启动httpd服务的镜 像 步骤 1. 创建⼀个⽬录&#xff0c;⽤于存储Docker file所使⽤的⽂件 1. 在此⽬录中创建Docker file⽂件&#xff0c;以及镜像制作所使⽤的⽂件 1. 使⽤docker build创建镜像 1. 使⽤创建的镜像…

数学建模学习(123):使用Python实现ARAS方法进行多准则决策实战

文章目录 1 引言1.1 简介1.2 ARAS方法的基本原理2. 案例分析:设备采购决策2.1. 数据集的构建2.2 权重的设定2.3. 准则类型的设定2.4. ARAS方法实现2.5. 结果分析1 引言 1.1 简介 多准则决策分析(MCDA)是一种用于帮助决策者在多个备选方案和多个相互冲突的准则中选择最佳方…

Golang学习总结

一、基本语法 Golang 设计理念&#xff1a;一种事情有且只有一种方法完成 软件包安装 直接官网下载好&#xff0c;配置安装下环境变量即可 /etc/profile # 在这个文件写入会对所有用户生效 ~/.profile # 在这个文件写入只会对当前用户生效# 以上两个文件中…

HarmonyOS NEXT未成年人模式无缝联动所有应用,过滤非适龄内容

背景 随着消费电子产品和移动互联网的普及&#xff0c;未成年人互联网普及率96.8%&#xff0c;超过80%的未成年人都拥有自己的上网设备&#xff0c;而如何引导孩子正确上网一直是家长们的担忧。市场上很多电子设备、系统推出了一些未成年人管控能力&#xff0c;却需要家长到各…

进程、线程的区别

进程&#xff08;Process&#xff09;和线程&#xff08;Thread&#xff09;是操作系统中的基本概念&#xff0c;它们在资源管理和任务执行方面有着本质的区别&#xff1a; 定义&#xff1a; 进程&#xff1a;进程是操作系统进行资源分配和调度的一个独立单位。每个进程都有自己…