在tensorflow中实现自定义op,在引用时有两种方式:
(1)将自定义实现的op编译成.so文件,在python测试代码中直接引用即可
(2)在python的user_ops中定义,这样不用再python测试代码中引入,调用起来更加方便
两种实现方式如下所示:
方式一:编译成.so文件
在tensorflow中实现自定义op,在引用时有两种方式:
(1)将自定义实现的op编译成.so文件,在python测试代码中直接引用即可
(2)在python的user_ops中定义,这样不用再python测试代码中引入,调用起来更加方便
两种实现方式如下所示:
方式一:编译成.so文件