笔记关键词检索?

在所有笔记中搜索你感兴趣的关键词!

《TensorFlow深度学习算法原理与编程实战》

作者:蒋子阳


tf.get_collection()

作用从一个集合中取出变量

tf.get_collection(
    key,
    scope=None
)

该函数可以用来获取key集合中的所有元素,返回一个列表。列表的顺序依变量放入集合中的先后而定。scope为可选参数,表示的是名称空间(名称域),如果指定,就返回名称域中所有放入‘key’的变量的列表,不指定则返回所有变量。

例子

variables = tf.get_collection(tf.GraphKeys.VARIABLES)
for i in variables:
  	print(i)

 

评论 (0)

发布评论

你的邮箱地址不会被公开。请输入所有带 * 的信息。