MobulaOP:简单灵活的跨框架操作员工具包 源码

MobulaOP Linux 视窗 覆盖范围 徽章 它是什么? MobulaOP是一个简单而灵活的跨框架操作员工具包。 您可以通过Python / C ++ / C / CUDA / HIP / TVM编写自定义运算符,而无需从源代码重建深度学习框架。 如何使用它? [] [] 添加加法运算符[] import mobula @ mobula . op . register class MyFirstOP : def forward ( self , x , y ): return x + y def backward ( self , dy ): return [ dy , dy ] def infer_shape ( self , in_shape ): assert in_shape [ 0 ]