最近项目,需要创建一个 tensorflow 的一个自定义操作,用来加速tensorflow的处理效果;下面对创建过程中,遇到的问题和资源进行简要记录,进行备忘:
OP 创建 参考链接:
https://www.tensorflow.org/guide/create_op
何时定义一个新的OP:
- 现有的operation 组合不出来需要的OP;
- 现有的operation 组合出来的OP 十分低效;
- 你想要融合一些操作来提高效率;
最近项目,需要创建一个 tensorflow 的一个自定义操作,用来加速tensorflow的处理效果;下面对创建过程中,遇到的问题和资源进行简要记录,进行备忘:
OP 创建 参考链接:
https://www.tensorflow.org/guide/create_op
何时定义一个新的OP: