如何在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)