学习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)