将sess.run输出放入lis

2024-04-19 16:20:36 发布

您现在位置:Python中文网/ 问答频道 /正文

所以我的代码中有一行代码,它可以工作:

    _, loss_value0, loss_value1, loss_value2, loss_value3 = sess.run([train_op, loss0, loss1, loss2, loss3])

我想把这列为损失清单,所以我跑了:

    loss_value_list = []
    _, loss_value_list = sess.run([train_op] + loss_list)

但这行不通。也许我不能分配这样的名单?有办法吗

谢谢你的帮助


Tags: run代码valuetrainlistsessopvalue1
1条回答
网友
1楼 · 发布于 2024-04-19 16:20:36

不需要预先定义loss_value_list;使用*运算符将结果解压到列表变量中:

lst = ['a', 1, 2, 3, 4]
_, *loss_value_list = lst

loss_value_list
# [1, 2, 3, 4]

相关问题 更多 >