如何在PyTorch中的[r1,r2]范围内获得均匀分布？

Question

问题就是这一切.我希望得到一个2-D torch.Tensor,其大小[a,b]填充了[r1,r2]PyTorch中均匀分布(范围内)的值.

Answer 1

如果U是在[0,1]上均匀分布的随机变量,则(r1 - r2) * U + r2均匀分布在[r1,r2]上.

因此,您只需要:

(r1 - r2) * torch.rand(a, b) + r2

Answer 2

torch.FloatTensor(a, b).uniform_(r1, r2)

Answer 3

利用该torch.distributions包生成来自不同发行版的样本。

[a,b]例如，要从均匀分布中采样大小为 2d PyTorch 张量，range(low, high)请尝试以下示例代码

import torch
a,b = 2,3   #dimension of the pytorch tensor to be generated
low,high = 0,1 #range of uniform distribution

x = torch.distributions.uniform.Uniform(low,high).sample([a,b])