关于注册自定义算子的标准流程,建议直接看Pytorch官方文档 英文教程:https://docs.pytorch.org/tutorials/advanced/cpp_custom_ops.html 中文教程:https://docs.pytorch.ac.cn/tutorials/advanced/cpp_custom_ops.html 对应github项目:https://github.com/pytorch/extension-cpp
关于注册自定义算子的标准流程,建议直接看Pytorch官方文档 英文教程:https://docs.pytorch.org/tutorials/advanced/cpp_custom_ops.html 中文教程:https://docs.pytorch.ac.cn/tutorials/advanced/cpp_custom_ops.html 对应github项目:https://github.com/pytorch/extension-cpp