在Tensorflow中解釋`tf.gather_nd`

Question

您可以直觀地解釋或給出有關tf.gather_nd的更多示例，用於索引和切入Tensorflow中的高維張量？我閱讀API，但它保持相當簡潔，我發現自己很難遵循該功能的概念。

謝謝。

Answer 1

好了，想想這樣的：

您所提供的索引值的列表索引所提供的由張來獲得這些值。索引的第一個維度是針對您要執行的每個索引。讓我們假裝張量只是一個列表清單。

[[0]]表示您想要在提供的張量中的索引0處返回一個特定值。就像這樣：

[tensor[0]] 

[0]，[1]指你想回到索引0和兩個具體數值1是這樣的：

[tensor[0], tensor[1]] 

現在什麼比如果張量是比較一個尺寸是多少？我們做同樣的事情：

[0,0]意味着你要在尺寸0的指數0回一個值，那麼維1.指數0，像這樣：

[tensor[0][0]] 

[0 ，1]，[2，3]]表示您希望在提供的索引和維度中返回兩個值。像這樣：

[tensor[0][1], tensor[2][3]] 

我希望這是有道理的。我嘗試使用Python索引來幫助解釋如何在Python中查看列表。