小编use*_*923的帖子

Cython 在 cdef 中返回元组?

您好,我正在尝试将 python 代码转换为 cython 以加快其计算速度。我试图在 cython 代码中从 a 返回多个cdef数组cpdef。基于经典 C,我可以使用指针或元组。我决定使用元组,因为大小不同。我知道下面的代码不起作用,有什么帮助吗?谢谢你!

import numpy as np
cimport numpy as np

cdef tuple funA(double[:] X, double[:] Y): 
    cdef int nX, nY, i
    nX = len(X)
    nY = len(Y)

    for i in range(nX):
        X[i] = X[i]*X[i]

    for i in range(nY):
        Y[i] = Y[i]*Y[i]

    return X,Y

cpdef Run(double[:] X, double[:] Y)
    cdef Tuple1, Tuple2 = funA(X,Y)

    # Do some calculation with Tuple1 and Tuple2 
    # Example

    cdef int i, nTuple1, nTuple2 …
Run Code Online (Sandbox Code Playgroud)

cython

4
推荐指数
1
解决办法
1万
查看次数

标签 统计

cython ×1