亚洲乱码中文字幕综合,中国熟女仑乱hd,亚洲精品乱拍国产一区二区三区,一本大道卡一卡二卡三乱码全集资源,又粗又黄又硬又爽的免费视频

pybind11和numpy進行交互的方法

 更新時間:2019年07月04日 10:26:08   作者:醬油  
這篇文章主要介紹了pybind11和numpy進行交互的方法,本文通過實例代碼給大家介紹的非常詳細,具有一定的參考借鑒價值 ,需要的朋友可以參考下

使用一個遵循buffer protocol的對象就可以和numpy交互了.

這個buffer_protocol要有哪些東西呢? 要有如下接口:

struct buffer_info {
  void *ptr;
  ssize_t itemsize;
  std::string format;
  ssize_t ndim;
  std::vector<ssize_t> shape;
  std::vector<ssize_t> strides;
};

其實就是一個指向數(shù)組的指針+各個維度的信息就可以了. 然后我們就可以用指針+偏移來訪問數(shù)字中的任意位置上的數(shù)字了.

下面是一個可以跑的例子:

#include <pybind11/pybind11.h>
 #include <pybind11/numpy.h>
 namespace py = pybind11;
 py::array_t<double> add_arrays(py::array_t<double> input1, py::array_t<double> input2) {
   py::buffer_info buf1 = input1.request(), buf2 = input2.request();
   if (buf1.ndim != 1 || buf2.ndim != 1)
     throw std::runtime_error("Number of dimensions must be one");
   if (buf1.size != buf2.size)
     throw std::runtime_error("Input shapes must match");
   /* No pointer is passed, so NumPy will allocate the buffer */
   auto result = py::array_t<double>(buf1.size);
   py::buffer_info buf3 = result.request();
   double *ptr1 = (double *) buf1.ptr,
      *ptr2 = (double *) buf2.ptr,
      *ptr3 = (double *) buf3.ptr;
   for (size_t idx = 0; idx < buf1.shape[0]; idx++)
     ptr3[idx] = ptr1[idx] + ptr2[idx];
   return result;
 }
 
 PYBIND11_MODULE(test, m) {
   m.def("add_arrays", &add_arrays, "Add two NumPy arrays");
 }

array_t里的buf就是一個兼容的接口.

buf中可以得到指針和對應數(shù)字的維度信息.

為了方便我們甚至可以使用Eigen當作我們兼容numpy的接口:

#include <pybind11/pybind11.h>
 #include <pybind11/eigen.h> 
 #include <Eigen/LU> 
 // N.B. this would equally work with Eigen-types that are not predefined. For example replacing
 // all occurrences of "Eigen::MatrixXd" with "MatD", with the following definition:
 //
 // typedef Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> MatD;
 
 Eigen::MatrixXd inv(const Eigen::MatrixXd &xs)
 {
  return xs.inverse();
 }
 
 double det(const Eigen::MatrixXd &xs)
 {
  return xs.determinant();
 }
 
 namespace py = pybind11;
 
 PYBIND11_MODULE(example,m)
 {
  m.doc() = "pybind11 example plugin";
 
  m.def("inv", &inv);
 
  m.def("det", &det);
 }

更多參考:

https://pybind11.readthedocs.io/en/stable/advanced/pycpp/numpy.html

https://github.com/tdegeus/pybind11_examples

總結(jié)

以上所述是小編給大家介紹的pybind11和numpy進行交互的方法,希望對大家有所幫助,如果大家有任何疑問請給我留言,小編會及時回復大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持!
如果你覺得本文對你有幫助,歡迎轉(zhuǎn)載,煩請注明出處,謝謝!

相關文章

  • python編程學習np.float 被刪除的問題解析

    python編程學習np.float 被刪除的問題解析

    這篇文章主要為大家介紹了python編程學習np.float 被刪除的問題解析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪
    2023-02-02
  • 淺談tensorflow與pytorch的相互轉(zhuǎn)換

    淺談tensorflow與pytorch的相互轉(zhuǎn)換

    本文主要介紹了簡單介紹一下tensorflow與pytorch的相互轉(zhuǎn)換,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧
    2022-06-06
  • 詳解用Python練習畫個美隊盾牌

    詳解用Python練習畫個美隊盾牌

    這篇文章主要介紹了用Python練習畫個美隊盾牌,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧
    2019-03-03
  • python查找目錄下指定擴展名的文件實例

    python查找目錄下指定擴展名的文件實例

    這篇文章主要介紹了python查找目錄下指定擴展名的文件,實例分析了Python文件查詢的技巧,非常具有實用價值,需要的朋友可以參考下
    2015-04-04
  • python ipset管理 增刪白名單的方法

    python ipset管理 增刪白名單的方法

    今天小編就為大家分享一篇python ipset管理 增刪白名單的方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
    2019-01-01
  • 解決pycharm安裝后代碼區(qū)不能編輯的問題

    解決pycharm安裝后代碼區(qū)不能編輯的問題

    今天小編就為大家分享一篇解決pycharm安裝后代碼區(qū)不能編輯的問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
    2018-10-10
  • Python Nose框架編寫測試用例方法

    Python Nose框架編寫測試用例方法

    這篇文章主要介紹了Python Nose框架編寫測試用例方法,需要的朋友可以參考下
    2017-10-10
  • python通過socket查詢whois的方法

    python通過socket查詢whois的方法

    這篇文章主要介紹了python通過socket查詢whois的方法,涉及Python基于socket模塊進行查詢的相關技巧,具有一定參考借鑒價值,需要的朋友可以參考下
    2015-07-07
  • 基于Python實現(xiàn)新年倒計時

    基于Python實現(xiàn)新年倒計時

    眼看馬上春節(jié)就要來臨了,所以滿懷期待的寫了一個Python新年倒計時的小工具!文中的示例代碼簡潔易懂,感興趣的小伙伴可以跟隨小編一起學習一下
    2023-01-01
  • python實現(xiàn)冒泡排序算法的兩種方法

    python實現(xiàn)冒泡排序算法的兩種方法

    本篇文章主要介紹了python實現(xiàn)冒泡排序的兩種方法,小編覺得挺不錯的,現(xiàn)在分享給大家,也給大家做個參考。一起跟隨小編過來看看吧
    2018-03-03

最新評論