本文共 1710 字,大约阅读时间需要 5 分钟。
以前每次看博客总有大神贴出一大堆源码来解析,今天突然就想写一个关于怎么找tensorflow源码的文章,这里的源码我觉得是来源于调用的库,而调用的库肯定是在本地有存档,所以,找源码就是找tensorflow中的目录~
论找函数源码的作用的话,我觉得首先就是源码中的注释很清晰,可以帮助搞懂函数参数的意义
比如我要找实现rnn中一个函数的源码
进入安装tensorflow的路径,如果对tensorflow设置了环境,anaconda里会有一个envs的文件夹,
1. 比如要找dynamic_rnn的实现源码
进入这个文件夹以后,有这些文件
打开rnn.py,搜索dynamic_rnn就有下面的这个,解释了函数的参数
2.找tf.contrib.rnn.BasicLSTMCell的实现源码
在rnn_cell_impl.py文件中,路径同上
分割线
————————————————————————————————————————————————————
最近看了很多大神的博客和知乎,哇,深觉分享真是一种美德。
之前手动找源码,实在找的太痛苦,主要痛苦在于不知道那个函数的实现在哪个文件当中,搞得我想要ctrl+F都不行,因此从网上找了一点代码,又自己加工了一小下,成功成为了找源码的好工具。
#或许还有其他更简便的方法我不知道的,请别见怪-_-
import osimport sysfindCount = 0findId = "bidirectional_dynamic_rnn"findDir = "C:/Users/dell/Anaconda3/envs/tensorflow1/Lib"resultDir = "" #自己写一个目录resultFile = os.path.join(resultDir,"bilstm.txt")def writeResultAndPrint(fullPath): file = open(resultFile,'a') file.write(fullPath) file.write("\n") print "write ok!\n" file.close()def findKey(findId,fullPath): file = open(fullPath,'r') content = file.read() file.close() isExist = content.find(findId) if isExist > 0: global findCount findCount = findCount + 1 writeResultAndPrint(fullPath)def findFiles(): clean() for dirPath,dirNames,fileNames in os.walk(findDir): for file in fileNames: fullPath = os.path.join(dirPath,file) findKey(findId,fullPath) print "has the string!!:" + str(findCount)def clean(): if os.path.exists(resultFile): os.remove(resultFile)if __name__ == '__main__':# a = sys.argv[1:]# findId += a[0]# findDir += a[1]# resultDir += a[2] findFiles()
findId就是要找的函数,findDir是你预计文件会在的目录,resultDir是结果存放的目录。
成功的截图:
虽然仍然是个笨方法,但还是比手动找简单多了,劳动工具解放生产力-_-