tensorflow-collection收集器

标签：oat   收集器   退出   收集   auth   进程   ESS   返回   增加   

get_collection得到collection中关于key的内容，返回值是一个列表。

add_to_collection将值增加到collection中，并与name关联

add_to_collections将一个值放入多个collection中

tf.get_collection(
key,
scope=None
)
tf.add_to_collection(
name,
value
)

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Mon Aug 27 11:16:32 2018

@author: myhaspl
"""

import tensorflow as tf
a = tf.constant([[1., 2.], [3., 4.]])
b = a+1
c= b+2

tf.add_to_collection("my_constant",a)
tf.add_to_collection("my_constant",b)  
tf.add_to_collection("my_constant",c)

const_vals=tf.get_collection("my_constant")
res=tf.add_n([a,b,c])                                                         
sess=tf.Session()
with sess:
    print sess.run(res)
    print sess.run(const_vals)

[[ 7. 10.]
[13. 16.]]
[array([[1., 2.],
[3., 4.]], dtype=float32), array([[2., 3.],
[4., 5.]], dtype=float32), array([[4., 5.],
[6., 7.]], dtype=float32)]
记住，上次跑程序已经保存的值不会被自动删除，值会累计保存，直到python进程退出。否则，再跑一次上面的程序，则

[[ 7. 10.]
[13. 16.]]
[array([[1., 2.],
[3., 4.]], dtype=float32), array([[2., 3.],
[4., 5.]], dtype=float32), array([[4., 5.],
[6., 7.]], dtype=float32), array([[1., 2.],
[3., 4.]], dtype=float32), array([[2., 3.],
[4., 5.]], dtype=float32), array([[4., 5.],
[6., 7.]], dtype=float32)]

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Mon Aug 27 11:16:32 2018

@author: myhaspl
"""

import tensorflow as tf
a = tf.constant([[1., 2.], [3., 4.]])

tf.add_to_collections(["my_con1","my_con2"],a)

const_1=tf.get_collection("my_con1")
const_2=tf.get_collection("my_con2")

sess=tf.Session()
with sess:

    print sess.run(const_1)
    print sess.run(const_2)

[array([[1., 2.],
? ? ? ?[3., 4.]], dtype=float32)]
[array([[1., 2.],
? ? ? ?[3., 4.]], dtype=float32)]

tensorflow-collection收集器

标签：oat   收集器   退出   收集   auth   进程   ESS   返回   增加   

原文地址：http://blog.51cto.com/13959448/2326091