python - 如何将 3d 和 4d numpy.array 传递给 swig

Question

我正在使用 python 来计算一些东西，但我想让它更快。所以我用了大口喝。

我想在同一个函数中使用 3d 数组和 4d 数组。

痛饮

%apply (double *INPLACE_ARRAY3, int DIM1, int DIM2, int DIM3)\
{(double *trans, int trans_dim1, int trans_dim2, int trans_dim3)};

%apply (double *INPLACE_ARRAY4, int DIM1, int DIM2, int DIM3, int DIM4)\
{(double *sample,int sam_dim1, int sam_dim2, int sam_dim3, int sam_dim4)};

样本.h

void update_transition(double *trans,int trans_dim1,int trans_dim2, int trans_dim3,
 double *sample,int sam_dim1, int sam_dim2, int sam_dim3, int sam_dim4, double DENO);

但是当我在 python 中使用它时，出现错误:

**TypeError: update_transition() takes exactly 7 arguments (3 given)**

这是否意味着我的 3d 数组可以被 swig 识别，但 4d 数组不能？如何解决这个问题呢？我两个都想要。

Answer 1

Python 表示函数调用需要 7 个参数。 C++ 函数期望 10。如果 SWIG 能够应用这两个类型映射，则 Python 函数期望 3，而不是 7。从 10 到 7 的唯一方法是 SWIG 错过第二个类型映射；因此它能够匹配函数的第一个数组，但不能匹配第二个数组。要确认这一点，请使用 10.3.6 Debugging typemap pattern matching 中提到的一些技术。 。另请查看生成的代码。您可能想要尝试的一件事是 5 参数 C++ 测试函数，它接受一个 1D 数组和一个 2D 数组，使用 INPLACE_ARRAY1 和 2 导出为 2 参数 Python 函数。这是一个更容易解决的案例。

如果这不能解决问题但为您提供了有用的信息，请将其添加到您的问题中，也许我或其他人可以再看一下。