在GPU上生成随机数-PyTorch

https://discuss.pytorch.org/t/random-number-on-gpu/9649/2

等价关系:

torch.rand(10, 10) 
torch.FloatTensor(10, 10).uniform_()

torch.randn(10, 10)
torch.FloatTensor(10, 10).normal_()

torch.cuda.FloatTensor(10, 10).normal_() 就会在gpu上生成正太分布的随机数。

你可能感兴趣的:(在GPU上生成随机数-PyTorch)