Cython Memoryview中的数组广播

Question

Cython Memoryview中的数组广播

numpycythonarray-broadcastingtyped-memory-views

4

我在Cython中创建了一个类型化的内存视图，并希望将其乘以一个标量：

import numpy as np
import math
cimport numpy as np

def foo():
    N = 10
    cdef np.double_t [:, :] A = np.ones(shape=(N,N),dtype=np.double_)
    cdef int i,j
    cdef double pi = math.pi
    for i in range(N):
        for j in range(N):
            A[i,j] *= pi
    return A

def bar():
    N = 10
    cdef np.double_t [:, :] A = np.ones(shape=(N,N),dtype=np.double_)
    cdef double pi = math.pi
    A *= pi
    return A

函数foo()可以完成这个任务，但不是很方便/易读。

然而，在函数bar()中的代码行A *= pi无法编译：Invalid operand types for '*' (double_t[:, :]; double)。

是否有一种方法在Cython内存视图上执行此类广播操作？

- David

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- DavidW · Accepted Answer

不，内存视图并不会这样做。内存视图只是一种快速访问数组单个元素的方法，它没有真正的概念可用于数组上可以执行的数学运算。

在您的“bar”函数中，任何尝试类型化它的操作可能实际上会使它变得更糟（即它将花费额外的时间来检查类型，但最终仍需使用普通调用Numpy函数完成工作）。

有许多（不完全令人满意的）方法从内存视图中获取一个Numpy数组：

np.asarray(memview) - this should be done without copying (provided you aren't using the esoteric indirect memory layout). It might be worth adding an assertion to check that no copy was made though.
memview.base - be slightly careful with this. If the memoryview is a result of slicing then .base will be the original unsliced object.
Keep a parallel numpy array and memoryview variable:
```
 Anp = np.array(...)
 cdef double[:] Amview = Anp
```
because the memoryview is a view of some memory, modifications to the array will be reflected in the memoryview and vice-versa. (Reassigning the array variable, e.g. Anp = something_else, won't be reflected though).

简而言之，memoryviews 的主要功能是快速访问单个元素。如果您不需要这样做，那么您可能不需要使用 memoryviews。