python - 如何将numpy数组转换为向量& (参考)与 SWIG
问题描述
我的目标:
在 python 中创建 3 个 numpy 数组(其中 2 个将使用特定值初始化),然后通过 swig 将所有三个数组作为向量引用发送到 c++ 函数中(这是为了避免复制数据并降低效率)。一旦进入 c++ 函数,添加 2 个数组并将它们的总和放入第 3 个数组。
vec_ref.h
#include <vector>
#include <iostream>
void add_vec_ref(std::vector<int>& dst, std::vector<int>& src1, std::vector<int>& src2);
vec_ref.cpp
#include "vec_ref.h"
#include <cstring> // need for size_t
#include <cassert>
void add_vec_ref(std::vector<int>& dst, std::vector<int>& src1, std::vector<int>& src2) {
std::cout << "inside add_vec_ref" << std::endl;
assert(src1.size() == src2.size());
dst.resize(src1.size());
for (size_t i = 0; i < src1.size(); i++) {
dst[i] = src1[i] + src2[i];
}
}
vec_ref.i
%module vec_ref
%{
#define SWIG_FILE_WITH_INIT
#include "vec_ref.h"
%}
%include "numpy.i"
%init %{
import_array();
%}
%include "std_vector.i"
%template(vecInt) std::vector<int>;
// %template(vecIntRef) std::vector<int> &;
// %apply (std::vector<int> * INPLACE_ARRAY1, int DIM1) {(std::vector<int> * dst, int a),(std::vector<int> * src1, int b),(std::vector<int> * src2, int c)};
// %apply (std::vector<int> * INPLACE_ARRAY1) {(std::vector<int> * dst),(std::vector<int> * src1),(std::vector<int> * src2)};
// %apply (std::vector<int> & INPLACE_ARRAY1) {(std::vector<int> & dst),(std::vector<int> & src1),(std::vector<int> & src2)};
// %apply (std::vector<int> & INPLACE_ARRAY1, int DIM1) {(std::vector<int> & dst, int a),(std::vector<int> & src1, int b),(std::vector<int> & src2, int c)};
%include "vec_ref.h"
生成文件
all:
rm -f *.so *.o *_wrap.* *.pyc *.gch vec_ref.py
swig -c++ -python vec_ref.i
g++ -O0 -g3 -fpic -c vec_ref_wrap.cxx vec_ref.h vec_ref.cpp -I/home/lmckeereid/tools/anaconda3/pkgs/python-3.7.3-h0371630_0/include/python3.7m/
g++ -O0 -g3 -shared vec_ref_wrap.o vec_ref.o -o _vec_ref.so
测试器.py
import vec_ref as vec
import numpy as np
a = np.array([1,2,3], dtype=np.intc)
b = np.array([4,5,6], dtype=np.intc)
c = np.zeros(len(a), dtype=np.intc)
print('---Before---\na:', a)
print('b:', b)
print('c:', c)
vec.add_vec_ref(c,a,b)
print('---After---\na:', a)
print('b:', b)
print('c:', c)
输出:
---Before---
a: [1 2 3]
b: [4 5 6]
c: [0 0 0]
Traceback (most recent call last):
File "tester.py", line 12, in <module>
vec.add_vec_ref(c,a,b)
TypeError: in method 'add_vec_ref', argument 1 of type 'std::vector< int,std::allocator< int > > &'
我已经尝试了在 vec_ref.i 中找到的所有已注释掉的 %apply 和 %template 指令,但它们都不起作用。
是否有一些我不应该包含的类型图?
解决方案
我同意@pschill:不复制数据就无法获得 std::vector 。
一种替代方法是使用std::span
类模板(在 C++20 中引入),或在库中定义的类似span
类模板。
创建 astd::span<int>
将提供数组中现有数据的视图,并在 C++ 中提供许多方便的成员函数(例如、迭代器、、等)。 numpy
operator[]
front()
back()
创建跨度永远不会从 numpy 数组中复制数据。
推荐阅读
- python - 如何将 .py 放入 .exe?
- typescript - npm typescript-language-server 错误:EACCES:权限被拒绝,符号链接
- javascript - 如何使用 HTML 文件在电子邮件中发送正确的日期
- python - 如何获取 tkinter 窗口的屏幕截图
- reactjs - React.js:条件渲染不加载子组件
- video - 如何使用ffmpeg在每10秒后剪切2秒的视频?
- .net - 无法更改状态栏背景色
- video - FFMPEG,播放器在编码后播放生涩的视频
- javascript - 将更新元素从一个文件反应到另一个文件
- python-3.x - Azure Python 函数错误--grpcMaxMessageLength 2147483647