numba并行化网格

Question

我正在寻找一种方法来重写下面的代码，以便numba可以使用它。当前，运行代码会提供错误Use of unsupported NumPy function 'numpy.meshgrid' or unsupported use of the function.。本质上，我是在计算三进制的Catesian积（例如[0 ... 13] x [0..13] x [0..13]）。

@numba.jit(nopython=True)
def sequences_of_length(repeat, n_values):
    a = list(range(0, n_values))
    c = [a] * repeat
    x = np.array(np.meshgrid(*c), dtype=np.int16).T.reshape(-1, repeat)
    return x

例如sequences_of_length(3, 13)应该输出：

[[ 0  0  0]
 [ 0  1  0]
 [ 0  2  0]
 ...
 [12 10 12]
 [12 11 12]
 [12 12 12]]
(2197, 3)

（其中2197 = 13 ^ 3）

Answer 1

我今天遇到了同样的问题，很不幸，我不得不发现mgrid不是numba支持的功能之一。 Here可以看到numba支持的支持的方法和功能。但是vstack()受支持，我认为下一步应该是vsatck？