我想过实现一个新的池层,如何将它作为一个新层添加到tensorflow中。 无法理解Tensorflow代码库。 如何开始的任何线索都会有所帮助。
答案 0 :(得分:0)
如果要为池添加新的tensorflow操作,可以查看:https://www.tensorflow.org/extend/adding_an_op。
您可能还想查看他们的github并添加拉取请求,如果有多人可以使用它:https://github.com/tensorflow/tensorflow/tree/master/tensorflow。