How to get a uniform distribution in a range [r1,r2] in PyTorch?

前端 未结 8 1652
深忆病人
深忆病人 2021-02-05 01:49

The question says it all. I want to get a 2-D torch.Tensor with size [a,b] filled with values from a uniform distribution (in range [r1,r2]

8条回答
  •  星月不相逢
    2021-02-05 02:27

    Please Can you try something like:

    import torch as pt
    pt.empty(2,3).uniform_(5,10).type(pt.FloatTensor)
    

提交回复
热议问题