將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]