如何使用GPU cuda.FloatTensor加速/ sub？

Question

我正在尝试加速PSO算法。我有两个问题，我希望能加快以下几点：

1）创造粒子;

2）运动方程式。

  

1）x = (xmax - xmin)*torch.rand(pop).dtype + xmin

     

2）vnext = w*torch.rand(pop).dtype + c1*torch.rand(pop).dtype*(b - x) + c2*torch.rand(pop).dtype(bg - x)

当我使用dtype = torch.cuda.FloatTensor时，使用dtype = torch.FloatTensor要慢一些。 如何使用GPU？

加快数学运算速度