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

在 PyTorch 中,你可以使用 torch.randinttorch.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.emptyrandom_

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

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

相关推荐
华清远见IT开放实验室3 分钟前
【每天学点AI】实战图像增强技术在人工智能图像处理中的应用
图像处理·人工智能·python·opencv·计算机视觉
OpenVINO 中文社区12 分钟前
实战精选|如何使用 OpenVINO™ 在 ElectronJS 中创建桌面应用程序
人工智能·openvino
只怕自己不够好16 分钟前
《OpenCV 图像缩放、翻转与变换全攻略:从基础操作到高级应用实战》
人工智能·opencv·计算机视觉
网络研究院22 分钟前
国土安全部发布关键基础设施安全人工智能框架
人工智能·安全·框架·关键基础设施
mqiqe27 分钟前
Elasticsearch 分词器
python·elasticsearch
YRr YRr34 分钟前
如何使用 PyTorch 实现图像分类数据集的加载和处理
pytorch·深度学习·分类
不去幼儿园2 小时前
【MARL】深入理解多智能体近端策略优化(MAPPO)算法与调参
人工智能·python·算法·机器学习·强化学习
想成为高手4992 小时前
生成式AI在教育技术中的应用:变革与创新
人工智能·aigc
YSGZJJ3 小时前
股指期货的套保策略如何精准选择和规避风险?
人工智能·区块链
无脑敲代码,bug漫天飞3 小时前
COR 损失函数
人工智能·机器学习