在运行python tensorflow时,要传递字符串或单个数字,我们可以使用tf.app.flags.DEFINE_string
或tf.app.flags.DEFINE_integer
。
但是我们可以将[2,3,4]或[7,8,9,6]之类的列表传递到python tensorflow吗?
答案 0 :(得分:1)
受到答案https://stackoverflow.com/a/24866869/5319143
的启发我使用tf.app.flags提出了类似的代码。
tf.app.flags.DEFINE_string("list_str", "3,4,5","list in string separated by comma")
the_list=[int(item) for item in FLAGS.list_str.split(',')]