python - 如何在运行 tensorflow 时使用命令行传递数字列表？

Question

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

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

标签： pythontensorflow

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