Как создать подкласс массива CuPy?

Question

Для NumPy можно использовать

class A(np.ndarray): pass
a = np.random.rand(10, 10).view(A)
print(a) # OK

Но для Cupy следующая операция приводит к segfault:

class A(cp.ndarray): pass
a = cp.random.rand(10, 10).view(A)
print(a) # segfault

Я просмотрел документ, но нашел мало информации.Это ожидается?Я использую CuPy 5.2, Python 3.6.8 и CUDA10.

Answer 1

Согласно документации numpy.ndarray.view , требуется два входных аргумента: dtype и type.Если аргумент dtype является подклассом numpy.ndarray, он интерпретируется как аргумент type.

С другой стороны, согласно документации cupy.ndarray.view , он принимает только один входной аргумент: dtype.Я думаю, cupy.ndarray.view не поддерживает type аргумент сейчас.