图解pytorch里面的torch.gather()

53 阅读 0 评论 35 点赞

我是靠谱客的博主迷路犀牛，最近开发中收集的这篇文章主要介绍图解pytorch里面的torch.gather()，觉得挺不错的，现在分享给大家，希望可以做个参考。

上图显示了 torch gather() 函数在 dim=1 的二维张量上的工作。

这里索引张量的行对应于输入张量的行（用灰色阴影突出显示）。现在对于索引张量中的每个索引值，从该行和输入张量的索引中选取相应的值。

让我们一步步理解这个例子——

代码实现如下所示——

在 [5] 中：

torch.gather(input=tensor2, dim=1, index = torch.tensor([[2, 1, 0], 
                                                         [0, 2, 1]]))

出[5]：

张量（[[2, 1, 0],
        [3, 5, 4]])

参考文献:

[Diagram] How to use torch.gather() Function in PyTorch with Examples - MLK - Machine Learning Knowledge

以上就是迷路犀牛为你收集整理的图解pytorch里面的torch.gather()的全部内容，希望文章能够帮你解决图解pytorch里面的torch.gather()所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。