簡體   English   中英

如何理解TensorFlow中的tf.get_collection()

[英]How to understand tf.get_collection() in TensorFlow

我很困惑tf.get_collection()形成文檔 ,它說

返回具有給定名稱的集合中的值列表。

這里有一個來自互聯網的例子

from_vars = tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES, from_scope)

這是否意味着它從tf.GraphKeys.TRAINABLE_VARIABLES收集變量到from_scope

但是,如果我想從另一個范圍獲取變量,我該如何使用此函數? 謝謝!

集合只不過是一組命名的值。

每個值都是計算圖的節點。

每個節點都有自己的名稱,名稱由范圍和/或值的串聯組成,例如: preceding/scopes/in/that/way/value

get_collection ,沒有scope允許在不應用任何過濾操作的情況下獲取集合中的每個值。

scope參數存在時,集合的每個元素都將被過濾,並且僅當節點的名稱以指定的scope開頭時才返回它。

如字符串doc中所述:

  • TRAINABLE_VARIABLES :將由優化程序訓練的Variable對象的子集。

scope :(可選。)一個字符串。 如果提供,則過濾結果列表以僅包含name屬性與scope匹配的項目,使用re.match 如果提供范圍,則永遠不會返回沒有name屬性的項。 re.match的選擇意味着沒有特殊標記的scope按前綴過濾。

因此它將返回給定范圍內的可訓練變量列表。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM