Rohit
Rohit

Reputation: 7161

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.

Upvotes: 3

Views: 3903

Answers (2)

Ivan
Ivan

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

Maleehak
Maleehak

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

Related Questions