将sess.run输出放入列表

Question

所以我的代码中有一行并且可以正常工作：

    _, 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)

但这不起作用。可能我无法分配这样的列表？有办法吗？

感谢您的帮助！

Answer 1

无需预先定义loss_value_list；使用*运算符将结果解压缩为列表变量：

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

loss_value_list
# [1, 2, 3, 4]