Я наивно пытаюсь индексировать в cuda поддерживаемый DeviceNDArray с массивом NumPy и получаю ошибку:
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 ()
. Для большего контекста я пытаюсь загрузить массивы с поддержкой cuda через механизм sklearn, чтобы мы смешали numpy / numba.
Возможно ли какое-либо индексирование?
Обновление
Под капотом DeviceNDArray вызывает Dim.__getitem__, который в настоящее время поддерживает только индексирование отдельных элементов и фрагментов