使用多个GPU加速numpy

Question

我有几个变量，我正在尝试执行以下操作：

np.array([ yHat[idS == i] for i in outIds])

此处，outIds，idS和yHat是相对较大的numpy数组。有没有办法获得相同的结果，但速度更快？我正在尝试为正确的ID获取yHat的值。订单很重要。我有多个GPU，所以多GPU解决方案也会很棒！