Question

如果我使用此功能

import numpy as np
from numba import jit

@jit(nopython=True)
def diss_matrix(data):
    n = data.shape[0]
    diss = np.empty((n, n))
    for i in range(n):
        for j in range(i):
            dist = np.absolute(data[i] - data[j]).sum()
            diss[i, j] = dist
            diss[j, i] = dist
    return diss

x = np.random.randn(100)
print(diss_matrix(x))

我收到此错误

numba.errors.UntypedAttributeError: Failed at nopython (nopython frontend)
Unknown attribute 'sum' of type float64
File "test_numba.py", line 11
[1] During: typing of get attribute at 
c:/Users/matte/Dropbox/Università/SDS/Thesis/source/test_numba.py (11)

我一直试图理解这意味着什么。事实是，触发错误的指令如下

dist = np.absolute(data[i] - data[j]).sum()

但我认为问题在于，numba假设data[i]和data[j]是float64而不是数组。实际上，以下代码

@jit(nopython=True)
def diss_matrix3():
    vec1 = np.array([1, 2, 3])
    vec2 = np.array([2, 3, 4])
    dist = np.absolute(vec1 - vec2).sum()
    return dist

完美无缺地工作。

我正在使用numba 0.35而我正试图找到一种方法来使该功能正常工作。我知道scipy.spatial.distance.pdist的存在，但我需要自己实现。此外，将来可能会出现同样的错误。

有什么建议吗？

Answer 1

如果你看np.random.randn(100)的形状是(100,)，那么data[i]确实是标量，而不是数组。如果您使用np.random.randn(100,100)，它应该可以使用。请查看docs for randn以获取有关函数如何工作的更详细说明。

与np.sum的Numba nopython错误

1 个答案: