Буферные объявления Cython для участников объекта

Я хочу иметь объект Cython cdef с членом NumPy и иметь возможность использовать быстрый доступ к буферам. В идеале я бы сделал что-то вроде:

import numpy as np
cimport numpy as np

cdef class Model:
  cdef np.ndarray[np.int_t, ndim=1] A

  def sum(self):
    cdef int i, s=0, N=len(self.A)
    for 0 <= i < N:
      s += self.A[i]
    return s

  def __init__(self):
    self.A = np.arange(1000)

К сожалению, Cython не может скомпилировать это с ошибкой Buffer types only allowed as function local variables.

Обходной путь, который я использую, заключается в объявлении атрибутов буфера новой локальной переменной, назначенной члену объекта:

cdef class Model:
  cdef np.ndarray A

  def sum(self):
    cdef int i, s=0, N=len(self.A)
    cdef np.ndarray[np.int_t, ndim=1] A = self.A
    for 0 <= i < N:
      s += A[i]
    return s

Это становится очень раздражающим, если вы хотите, чтобы несколько методов обращались к тем же структурам данных, что кажется довольно распространенным случаем использования, no?

Есть ли лучшее решение, которое не требует повторного объявления типов внутри каждого метода?

Ответы

import numpy as np
cimport numpy as np

  cdef class Model:

    cdef int [:] A

    def sum(self):

        for 0 <= i < N:
            s += self.A[i]
        return s

    def __init__(self):
        self.A = np.arange(1000)