当前位置: 首页> 科技> 名企 > pytorch初始化张量并填充随机整数值

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

时间:2025/7/11 1:04:28来源:https://blog.csdn.net/qq_27390023/article/details/141726424 浏览次数:2次

在 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

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

关键字:pytorch初始化张量并填充随机整数值

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

责任编辑: