Dropout can use torch.CudaTensor:bernoulli()