转至: https://blog.csdn.net/Suan2014/article/details/76269241
学习TensorFlow程序代码时,很多时候可以看到tf.app.run(),下面讲一下该函数的作用和用法。
1 函数原型
run(main=None, argv=None)
Runs the program with an optional 'main' function and 'argv' list.
2 作用
通常用此函数解析命令行后,运行main函数
3 例子
import tensorflow as tf
#导入命令行解析模块
import argparse
import sys
FLAGS=None
def main(_):
print(sys.argv[0])
if __name__=="__main__": #用这种方式保证了,如果此文件被其他文件import的时候,不会执行main中的代码
#创建对象
parse=argparse.ArgumentParser()
#增加命令行
parse.add_argument('--dataDir',type=str,default='\\tmp\\tensorflow\\mnist\\inputData',
help='Directory for string input data')
FLAGS, unparsed=parse.parse_known_args()
tf.app.run(main=main, argv=[sys.argv[0]] + unparsed) #解析命令行参数,调用main函数 main(sys.argv