tf.gather_nd直观上是做什么的?
tensorflow
4
0

您是否可以直观地解释或给出有关tf.gather_nd更多示例,以便在tf.gather_nd中索引和切片高维张量?

我阅读了API ,但简洁起见,我很难遵循该函数的概念。

参考资料:
Stack Overflow
收藏
评论
共 1 个回答
高赞 时间 活跃

好的,这样考虑:

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

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

[tensor[0]]

[[0],[1]]表示您想要在索引0和1处获得两个特定的切片,如下所示:

[tensor[0], tensor[1]]

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

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

[tensor[0][0]]

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

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

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

收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号