怎么在python3 中利用Eigen對(duì)代碼進(jìn)行加速

python -m venv env

./env/Scripts/Activate.ps1

pip install pybind11

 pip install numpy==1.19.3 

#include <Eigen/Dense>
using namespace std
using namespace Eigen
MatrixXd add_mat(MatrixXd A_mat, MatrixXd B_mat)
{
  return A_mat + B_mat;
}

// cpp_python.cpp : 此文件包含 "main" 函數(shù)。程序執(zhí)行將在此處開(kāi)始并結(jié)束。
//
#include <pybind11/pybind11.h>
#include <pybind11/eigen.h>
#include<pybind11/numpy.h>
#include<fstream>
#include<iostream>
#include <Eigen/Dense>
using namespace std;
using namespace Eigen;

MatrixXd add_mat(MatrixXd A_mat, MatrixXd B_mat)
{
	return A_mat + B_mat;
}

namespace py = pybind11;
PYBIND11_MODULE(add_mat_moudle, m)
{
	m.doc() = "Matrix add";//解釋說(shuō)明
	m.def("mat_add_py"/*在pyhon中使用的函數(shù)名*/, &add_mat);
}

from setuptools import setup
from setuptools import Extension

add_mat_module = Extension(name='add_mat_moudle', # 模塊名稱(chēng)
              sources=['cpp_python.cpp'],  # 源碼
              include_dirs=[r'.\eigen-3.3.8',
                    r'.\env\Scripts',   # 依賴(lài)的第三方庫(kù)的頭文件
                     r'.\env\Lib\site-packages\pybind11\include']
              )

setup(ext_modules=[add_mat_module])

執(zhí)行: "python .\setup.py build_ext --inplace"就會(huì)得下面的結(jié)果，生成.pyd文件表明我們已經(jīng)編譯成功。