tf 中定义了 tf.app.flags.FLAGS ,用于接受从终端传入的命令行参数
tf.app.flags.DEFINE_string() :定义一个用于接收 string 类型数值的变量;
tf.app.flags.DEFINE_integer() : 定义一个用于接收 int 类型数值的变量;
tf.app.flags.DEFINE_float() : 定义一个用于接收 float 类型数值的变量;
tf.app.flags.DEFINE_boolean() : 定义一个用于接收 bool 类型数值的变量;
“DEFINE_xxx”函数带3个参数,分别是变量名称,默认值,用法描述,例如:
import tensorflow as tf
flags = tf.app.flags
flags.DEFINE_string('first_num', '1',
'input first number')
flags.DEFINE_integer('second_num', 2,
'input second number')
FLAGS = flags.FLAGS
print(int(FLAGS.first_num)+FLAGS.second_num)
输出为3
在这里改变输入条件
python test_flags.py --first_num=5 --second_num=7
得到结果输出为
12