torch隨機數 manual_seed

import torch
seed = 2018
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
a=torch.rand([1,5])
# torch.manual_seed(seed)
# torch.cuda.manual_seed(seed)
b=torch.rand([1,5])
print(a,b)

按上面來得到的隨機數不同，加上注釋就會得到相同的隨機數。