在 numba 中，array(float64, 1d, C) 和 array(float64, 1d, A) 有什么区别？

Question

我有两个简单而相似的功能。一个可以用 numba 编译，而另一个不能。我无法理解 them.The 以下是这两个函数之间的区别：

第一个：

@nb.njit(float64[:](float64[:], float64[:]))
def arrAdd(a,b):
    assert a.shape == b.shape
    return a + b

编译成功。当我调用它时，

arrAdd(np.array([1,2.0,21]),np.array([2,3.0,1]))

会return:

array([ 3.,  5., 22.])

第二个：

c = np.array([1,2.0,21])
@nb.njit
def arrAdd1(arr):
    return arrAdd(arr,c)

然而，当我调用这个函数时：

arrAdd1([2,3.0,1])

它将显示：

TypingError: Failed in nopython mode pipeline (step: nopython frontend)
Invalid use of type(CPUDispatcher(<function arrAdd at 0x00000212A9FDC670>)) with parameters (array(float64, 1d, C), readonly array(float64, 1d, C))
Known signatures:
 * (array(float64, 1d, A), array(float64, 1d, A)) -> array(float64, 1d, A)
During: resolving callee type: type(CPUDispatcher(<function arrAdd at 0x00000212A9FDC670>))
During: typing of call at <ipython-input-57-c77e552c5560> (4)


File "<ipython-input-57-c77e552c5560>", line 4:
def arrAdd1(arr):
    return arrAdd(arr, c)
    ^

那么 array(float64, 1d, C)) 和 array(float64, 1d, A) 有什么区别呢？

Answer 1

A（any）、C（C-contiguous）和F（Fortran-contiguous）是array layout的三种类型。但这不是您示例中的问题。

问题 1

在这一行

arrAdd1([2,3.0,1])

您传递的是列表而不是数组。

以下简化版本有效：

@nb.njit                            # No types
def arrAdd(a, b):
    assert a.shape == b.shape
    return a + b

a = arrAdd(np.array([1, 2.0, 21]), np.array([2, 3.0, 1]))
print(a)

c = np.array([1, 2.0, 21])
@nb.njit                            # No types
def arrAdd1(arr):
    return arrAdd(arr, c)

a = arrAdd1(np.array([2,3.0,1]))    # Pass an array
print(a)

并生产

[ 3.  5. 22.]
[ 3.  5. 22.]

问题2

在您的示例中，arrAdd1() 被定义为闭包，因此 c 成为函数中的常量。

如果你真的要使用显式参数类型，你需要指定addArr()至少在第二个参数中接收一个常量数组。

只要函数不修改它们的输入，您就可以将所有输入参数声明为只读，如本例所示，它会产生相同的结果：

vector = nb.types.Array(dtype=f8, ndim=1, layout="A")
readonly_vector = nb.types.Array(dtype=f8, ndim=1, layout="A", readonly=True)

@nb.njit(vector(readonly_vector, readonly_vector))
def arrAdd(a, b):
    assert a.shape == b.shape
    return a + b

a = arrAdd(np.array([1, 2.0, 21]), np.array([2, 3.0, 1]))
print(a)

c = np.array([1, 2.0, 21])
@nb.njit(vector(readonly_vector))
def arrAdd1(arr):
    return arrAdd(arr, c)

a = arrAdd1(np.array([2,3.0,1]))
print(a)

您可以将布局（A、C、F）更改为最适合您的布局。

在 numba 中，array(float64, 1d, C) 和 array(float64, 1d, A) 有什么区别？

In numba, what's the difference between array(float64, 1d, C) and array(float64, 1d, A)?

python

arrays

numpy

numba