Hi,
You can use methods such as uniform_()
or normal_()
on cuda tensors to generate random numbers directly on gpu.
Keep in mind that torch.rand(10, 10)
is equivalent to torch.FloatTensor(10, 10).uniform_()
and torch.randn(10, 10)
is equivalent to torch.FloatTensor(10, 10).normal_()
.
So torch.cuda.FloatTensor(10, 10).normal_()
will do what you want.