《TensorFlow深度学习算法原理与编程实战》
作者:蒋子阳
在所有笔记中搜索你感兴趣的关键词!
作者:蒋子阳
作用:从一个集合中取出变量
tf.get_collection(
key,
scope=None
)
该函数可以用来获取key集合中的所有元素,返回一个列表。列表的顺序依变量放入集合中的先后而定。scope为可选参数,表示的是名称空间(名称域),如果指定,就返回名称域中所有放入‘key’的变量的列表,不指定则返回所有变量。
例子
variables = tf.get_collection(tf.GraphKeys.VARIABLES)
for i in variables:
print(i)
评论 (0)