3

我只是被 SWIG 和 numpy 的一个小问题困住了。我在 C 中有一个专门的矩阵 (n,k) 向量 (k) 产品，我想与 numpy 接口。所以我总是在乘法之前知道输出向量 (n) 的形状。

目前，C 函数分配内存并将 double* 返回给计算的乘积。为了防止内存泄漏并避免更改 C 代码，我像在点示例中一样对其进行接口：

%apply (int DIM1, int DIM2, double* IN_ARRAY2) {(int m1, int nd1, double* xd)};
%apply (int DIM1, double* IN_ARRAY1) {(int nd2, double* zd)};
%apply (int DIM1, double* ARGOUT_ARRAY1) {(int nd4, double* za)};

%rename (mat_vec_mul) my_mat_vec_mul;
%ignore mat_vec_mul;
%include "math_lib.h"
%exception my_mat_vec_mul {
    $action
    if (PyErr_Occurred()) SWIG_fail;
}
%inline %{
void my_mat_vec_mul( int m1, int nd1, double* xd, int nd2, double* zd, int nd4, double* za) 
{
    if (nd1 != nd2) {
        PyErr_Format(PyExc_ValueError,
                     "Arrays of lengths (%d,%d) given",
                     nd1, nd2);
    }

    int i;
    double *tmp = NULL;
    tmp = mat_vec_mul(m1,nd1,xd,zd);
    for( i = 0; i < nd1; i++){
        za[i] = tmp[i];
    } 
    free(tmp);
}
%}

现在，因为我将输出向量指定为 ARGOUT_ARRAY1，所以该函数在 python 中调用为

v = test.mat_vec_mul(A,b,A.shape[0])

这是不方便的。有没有办法在内部告诉 swig 的大小？或者是提供用户友好界面以在其周围添加额外包装器的唯一方法：

def user_interface_mat_vec_mul(A,b):
    mat_vec_mul(A,b,A.shape[0])

提前致谢。

4

1 回答 1

1

这是我的答案，我找到了一种将包装函数放入 swig 文件的方法，它基本上可以满足我的要求：

%pythoncode %{
def mat_vec_mul(A, b):
    return _test._mat_vec_mul(A, b, A.shape[0])
%}
于 2013-09-22T16:25:53.713 回答