Create an op on tensorflow; 在tensorflow 1.7/2.0 中创建一个 Op操作

最近项目,需要创建一个 tensorflow 的一个自定义操作,用来加速tensorflow的处理效果;下面对创建过程中,遇到的问题和资源进行简要记录,进行备忘:

OP 创建 参考链接:

https://www.tensorflow.org/guide/create_op 

何时定义一个新的OP:

  • 现有的operation 组合不出来需要的OP;
  • 现有的operation 组合出来的OP 十分低效;
  • 你想要融合一些操作来提高效率;

猜你喜欢

转载自www.cnblogs.com/xuyaowen/p/tensorflow-op-customize.html
OP
今日推荐