需要自定义某个层，或有时候用c++实现你的操作(c++扩展)可能会更好: 例如：需要实现一个新型的激活函数例如: bevfusion用cuda实现bevpool加速自定义扩展的步骤 (1) 首先用纯pytorch和python 实现我们所需的功能，看看效果再决定要不要进一步优化(2) 明确优化方向，用C++ (或CU…