求numba对np.stack的处理

Question

a = np.array([[1.0,2,3,4],
[5,6,7,8],
[9,10,11,12]])

b = [1,2,3]

@numba.njit(cache=True, parallel=True)
def add(a, b):
temp = []
for i in numba.prange(len(b)):
temp.append(a[i])
out = np.vstack(temp)
return out

ERROR expression:
numba.core.errors.TypingError: Failed in nopython mode pipeline (step: nopython frontend)
No implementation of function Function(<function vstack at 0x000002F49953E4C8>) found for signature:

vstack(list(array(float64, 1d, C))<iv=None>)

File "test.py", line 15:
def add(a, b):
<source />
temp.append(a[i])
out = np.vstack(temp)
^

Answer 1

首先no.vstack就不是这么用的，你这么堆叠出来的逻辑都不对，至少也是out=np.vstack((out,temp))。
第二，你的b只是提供length吗？
第三，为什么不再赋个temp_list=[]，直接append temp？然后在外部再用np.array()