Pytorch添加自定义算子之(2)-配置torch算子

一、安装配置pytorch环境

https://pytorch.org/get-started/locally/

二、安装eigen3库

见上一篇文章

三、编写CPP代码算子

将此cpp命名为:custom_group_norm.cpp,后续python 中会使用到。

#include 
#include "eigen3/Eigen/Dense"

using ConstEigenVectorArrayMap =<

你可能感兴趣的:(深度学习之自定义算子实现,pytorch,人工智能,python)