pytorchtensor

How to create a tensor of given shape and interval?

I am going through Pytorch and want to create a random tensor of shape 5X3 in the interval [3,7)

torch.rand(5,3) will return a random tensor of shape 5 X 3, however, I could not figure to set the given interval.

Please guide.

Solution

  • You can map u ~ U(0, 1) to U ~ [a, b] with u -> (b - a)*u + a:

    (b - a)*torch.rand(5, 3) + a