如何在boostnumpy库中使用boost::numpy::from_data

2024-06-23 03:16:19 发布

您现在位置:Python中文网/ 问答频道 /正文

我在用增强.numpy试着做一个包装纸。我尝试将ndarray从python发送到c++并将c++发送到python。当我从python向c++发送ndarray时,出现了问题。什么是不正确的?在

+++mod2.cpp+++
1 #include"boost/numpy.hpp"
2 #include<iostream>
3 
4 boost::numpy::ndarray test(boost::numpy::ndarray a){
5   boost::numpy::ndarray b = a;
6  int arr[]={5,4,3,2,1};
7   b =      boost::numpy::from_data(arr,boost::numpy::dtype::get_builtin<int>(),
8       boost::python::make_tuple(5),boost::python::make_tuple(sizeof(int)),boost::python::object());
9    std::cout << "In mymod2.cpp:" <<
10      boost::python::extract < char const * > (boost::python::str(b)) << std::endl ; 
11   return b;
12 }
13 
14 BOOST_PYTHON_MODULE(mymod2){
15   boost::numpy::initialize();
16   boost::python::def("test",test);
17 }               

+++demo.py+++
1 import numpy
2 import mymod2
3 
4 Mat1 = numpy.array([1,2,3,4,5], dtype=numpy.int32)
5 print("In demo.py:",Mat1)                                                                                         
6 Mat2=mymod2.test(Mat1)
7 print("In demo.py:",Mat2)

+++Output+++
In demo.py: [1 2 3 4 5]
In mymod2.cpp:[5 4 3 2 1]
In demo.py: [        0        48 430934944     32509         3]

Tags: inpytestnumpymakeincludedemocpp
3条回答

int arr[]={5,4,3,2,1}在堆栈上。from\u data引用传递给它的数据,它不复制它,所以在ndarray中返回堆栈地址。使用new意味着可以避免这个问题,因为内存在堆上,但是现在数据正在泄漏。在释放ndarry时,需要使用owner参数来管理内存的整理。在

在返回数组之前,可以创建对数组的新引用

np::ndarray new_b = b.copy();

我不明白为什么,但是如果你使用:

int *arr;
arr = new int[5];
arr[0] = 5;
arr[1] = 4;
arr[2] = 3;
arr[3] = 2;
arr[4] = 1;

它将返回正确的数组给python代码,而不是int arr[]={5,4,3,2,1}。在

相关问题 更多 >

    热门问题