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

Question

我天真地试图索引到一个带有NumPy数组的cuda支持的DeviceDarray，但出现了一个错误：

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: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

关于更多内容，我正在尝试通过sklearn机制加载cuda支持的数组，因此我们混合了numpy/numba。你知道吗

是否可以建立索引？你知道吗

更新

引擎盖下DeviceNDArray调用Dim.__getitem__，目前只支持单个项目和切片索引