Reputation: 7161
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.
Upvotes: 3
Views: 3903
Reputation: 40618
You can map u ~ U(0, 1)
to U ~ [a, b]
with u -> (b - a)*u + a
:
(b - a)*torch.rand(5, 3) + a
Upvotes: 1
Reputation: 801
Define the minimum and maximum value and use the code below:
import torch
max = 7
min = 3
rand_tensor = (max-min)*torch.rand((5, 3)) + min
Upvotes: 0