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)

按上面来得到的随机数不同，加上注释就会得到相同的随机数。