如何在运行tensorflow中使用命令行传递数字列表?

时间:2018-09-03 09:08:43

标签: python tensorflow

在运行python tensorflow时,要传递字符串或单个数字,我们可以使用tf.app.flags.DEFINE_stringtf.app.flags.DEFINE_integer

但是我们可以将[2,3,4]或[7,8,9,6]之类的列表传递到python tensorflow吗?

1 个答案:

答案 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(',')]