我正在尝试使用并加速花式索引以"连接"两个数组并对结果'轴之一求和.
像这样的东西:
$ ipython
In [1]: import numpy as np
In [2]: ne, ds = 12, 6
In [3]: i = np.random.randn(ne, ds).astype('float32')
In [4]: t = np.random.randint(0, ds, size=(1e5, ne)).astype('uint8')
In [5]: %timeit i[np.arange(ne), t].sum(-1)
10 loops, best of 3: 44 ms per loop
是否有一种简单的方法来加速声明In [5]？我应该去使用OpenMP和类似scipy.weave或Cython的prange？