load_op_library
class paddle.fluid.load_op_library
[源代码]
load_op_library
用于自定义C++算子中,用来加载算子动态共享库。加载库后,注册好的算子及其Kernel实现将在PaddlePaddle主进程中可以被调用。 请注意,自定义算子的类型不能与框架中的现有算子类型相同。
参数:
- lib_filename (str) – 动态共享库的名字。
代码示例
import paddle.fluid as fluid
#fluid.load_op_library('custom_op.so')