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)

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