无法从 TF 集合中获取变量

数据挖掘 机器学习 Python 神经网络 深度学习 张量流
2022-02-17 05:19:21

Aurelion Geron 的示例展示了如何将选定的操作存储在集合中,以便以后可以轻松访问它们:

添加:

for op in (X, y, accuracy, training_op):
    tf.add_to_collection("my_important_ops", op)

然后检索:

X, y, accuracy, training_op = tf.get_collection("my_important_ops")

但是,当我尝试这样做时,会出现此错误:

X, y, accuracy, training_op = tf.get_collection("my_important_ops")
ValueError: too many values to unpack (expected 4)

我究竟做错了什么?

1个回答

一个简单的解决方案是执行以下操作:

temp = tf.get_collection("my_important_ops")
print(temp.get_shape().as_list())

然后您可以评估输出。