BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #27133同步于 2017/12/6
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖

【问题】tensorflow添加new op,问题求教

BYD21
2017/12/6镜像同步2 回复
我打算在tensorflow实现一个自定义的OP,因为功能比较复杂,需要用c++去写,但是苦于tensorflow源码不精,遇到了一些问题。 1.大家有人用过clion或者其他IDE编译过tensorflow的源码吗?我用编辑器sublime去读框架core里的源码,效率太低了,希望能通过IDE提高下学习效率。 2.我现在Op的计算过程已经有c++实现了,首先就是怎么把指针类型的替换成tensorflow里要求的Tensor类。 3.有没有大神读过tensorflow的源码,私下交流一下呗。 最近被这个搞的快不行了,有种无从下手的感觉
订阅后,新回复会通过你的通知中心匿名送达。
2 条回复
qcts机器人#1 · 2017/12/8
确定需要C++去写?编程常见的循环和判断语句都有对应的TF函数来支持。由于早期TF的文档比较混乱,我稍微看过一点TF python部分的源码,很多TF自带的OP也就是用更加基础的python API搭建起来的。
BYD21机器人#2 · 2017/12/9
【 在 qcts 的大作中提到: 】 : 确定需要C++去写?编程常见的循环和判断语句都有对应的TF函数来支持。由于早期TF的文档比较混乱,我稍微看过一点TF python部分的源码,很多TF自带的OP也就是用更加基础的python API搭建起来的。 因为想要的op比较复杂,已经有caffe的c++实现版本,就像利用一下,而且也考虑过用python写,但是计算量太多,涉及到复杂的滤波过程,索性就拿c++写了。 编译通过了,也能在python里用了,在对这个op进行测试。