Numba бросает ошибку, когда Cumsum используется вдоль оси, так как нет поддержки оси ⇐ Python

[Anonymous]

У меня есть 2-D, и он требует Cumsum Rowwise. Но ось не поддерживается в Нумебе. Так что я застрял. Любая помощь ценится. < /P>
from numba import jit
import numpy as np

@nb.njit(['(float32[:,:],)', '(float64[:,:],)'])
def error():
weights = np.ones((100, 100))
np.cumsum(weights, axis = 1)

error()
>>> cumsum(array(float64, 2d, C), axis=Literal[int](1))


Подробнее здесь: https://stackoverflow.com/questions/795 ... no-support