Cython计算未知维度numpy数组维度和的编译错误排查与解决

错误原因 #

你用Cython声明的ndarray是底层的C语言类型对象，它的.shape属性返回的是C指针类型npy_intp *，而非Python的tuple对象。直接把C指针赋值给Python元组变量，Cython无法自动完成这种跨类型的转换，因此抛出编译错误。

解决建议 #

提供三种可行的修复方案，按需选择：

方案一：改用Python层面的numpy数组类型 #

放弃Cython的ndarray类型声明，直接用Python对象作为参数，让Cython自动处理numpy数组的Python接口：

import numpy as np

cpdef int dimsum(x):
    cdef int N
    N = np.sum(x.shape)
    return N

方案二：手动从C指针构建Python元组 #

如果需要保留Cython的ndarray类型，可借助numpy的C API获取维度信息，手动构建Python元组：

from numpy cimport ndarray, PyArray_NDIM, PyArray_SHAPE
import numpy as np

cpdef int dimsum(ndarray x):
    cdef:
        int N, i
        int ndim = PyArray_NDIM(x)
        npy_intp *shape_ptr = PyArray_SHAPE(x)
        tuple shape
    
    shape = tuple(shape_ptr[i] for i in range(ndim))
    N = np.sum(shape)
    return N

方案三：C层面直接计算（效率最优） #

如果只是计算维度的总和，完全不需要转换成Python元组，直接在C层面循环累加，性能更好：

from numpy cimport ndarray, PyArray_NDIM, PyArray_SHAPE

cpdef int dimsum(ndarray x):
    cdef:
        int total = 0, i
        int ndim = PyArray_NDIM(x)
        npy_intp *shape_ptr = PyArray_SHAPE(x)
    
    for i in range(ndim):
        total += shape_ptr[i]
    return total

内容的提问来源于stack exchange，提问作者Wei Li