torch.rand()、torch.randn()、torch.randint()、torch.randperm()用法

最新推荐文章于 2025-03-09 19:33:02 发布

徕胖

最新推荐文章于 2025-03-09 19:33:02 发布

阅读量10w+

点赞数 108

分类专栏： pytorch

本文链接：https://blog.csdn.net/leilei7407/article/details/107710852

版权

pytorch 专栏收录该内容

2 篇文章

订阅专栏

本文详细介绍了PyTorch中用于生成随机数的几个关键函数，包括torch.rand()用于生成均匀分布的浮点数，torch.randn()用于生成标准正态分布的浮点数，torch.randint()用于生成指定范围内的整数，以及torch.randperm()用于生成不重复的整数序列。这些函数在构建和训练神经网络模型时非常有用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch.rand和torch.randn有什么区别？ 
一个均匀分布，一个是标准正态分布。

torch.rand()
在这里插入图片描述

#rand(*size, out=None, dtype=None)
t1 = torch.rand(2,3)
print(t1,t1.type())

在这里插入图片描述
randn(*size, out=None, dtype=None)和randn_like(input, dtype=None)

#randn(*size, out=None, dtype=None)
#randn_like(input, dtype=None)
t1 = torch.randn(2,3)
t2 = torch.randn_like(t1)
print(t1,t1.type())
print(t2,t2.type())

在这里插入图片描述
torch.randint(),torch,randint_like()

#randint(low=0, high, size, out=None, dtype=None)
#randint_like(input, low=0, high, dtype=None)
#整数范围[low, high)
t1 = torch.randint(1,4,(2,3,2)) #形状写成[2,3,2]也行
t2 = torch.randint_like(t1,4)
print(t1)
print(t2)

在这里插入图片描述
torch.randperm()
randperm(n, out=None, dtype=torch.int64)-> LongTensor

#torch中没有random.shuffle
#y = torch.randperm(n) y是把1到n这些数随机打乱得到的一个数字序列
import torch
#randperm(n, out=None, dtype=torch.int64)-> LongTensor
idx = torch.randperm(3)
a = torch.Tensor(4,2)
print(a)
print(idx,idx.type())
print(a[idx])

在这里插入图片描述