Skip to main content
 首页 » 编程设计

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

2024年12月31日25飞鱼

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

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

请您参考如下方法:

灵感来自答案 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(',')]