将Numpy索引数组与Numba DeviceNDArray一起使用

Question

我很天真地尝试使用cuda数组索引到一个NumPy支持的DeviceNDArray并得到一个错误：

In [1]: import numpy as np

In [2]: import numba.cuda

In [3]: aa = numba.cuda.to_device(np.arange(10))

In [4]: aa[0]
Out[4]: 0

In [5]: aa
Out[5]: <numba.cuda.cudadrv.devicearray.DeviceNDArray at 0x7f6a647a07f0>

In [6]: aa[1]
Out[6]: 1

In [7]: aa[np.arange(5)]

  

ValueError：具有多个元素的数组的真值不明确。使用a.any（）或a.all（）

有关更多信息，我正在尝试通过sklearn机器加载cuda支持的数组，以便我们混合使用numpy / numba。

可以建立索引吗？

更新

内幕DeviceNDArray呼唤Dim.__getitem__，目前仅支持单个项目和切片索引