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, ...
確定! 回上一頁