python - 如何子类化 CuPy 数组？

Question

对于 NumPy，可以使用

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

但是对于 Cupy，以下操作会导致段错误：

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论点。