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

Question

这个问题说明了一切。我想得到一个尺寸为[a,b]的2-D torch.Tensor，其中填充了来自PyTorch中统一分布（范围为[r1,r2]）的值。

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)