tensorflow - tf.gather_nd直观上是做什么的？

Question

您能否直观地解释或给出有关tf.gather_nd的更多示例，以便在Tensorflow中对高维张量进行索引和 slice ？

我读了API，但是非常简洁，以至于我发现自己很难遵循该函数的概念。

Answer 1

好的，这样考虑:

您正在提供索引值列表，以对提供的张量进行索引以获取这些 slice 。您提供的索引的第一维是针对要执行的每个索引。让我们假设张量只是列表的列表。

[[0]]表示要在提供的张量的索引0处获得一个特定的 slice (列表)。像这样:

[tensor[0]]

[[0]，[1]]表示要在索引0和1处获得两个特定的 slice ，如下所示:

[tensor[0], tensor[1]]

现在，如果张量超过一个维度怎么办？我们做同样的事情:

[[0，0]]表示您想在第0个列表的索引[0,0]处获取一个 slice 。像这样:

[tensor[0][0]]

[[0，1]，[2，3]]表示您要以提供的索引和尺寸返回两个 slice 。像这样:

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

我希望这是有道理的。我尝试使用Python索引来帮助解释如何在Python索引表中执行此操作。