pytorch-(torch.take())根据索引返回指定索引上的数据集合

Posted 2022-07-14 wx62cecd4679aef

torch.take(input, index)->Tensor

返回一个新的张量，其中的元素是输入元素在给定的索引处，将输入张量视为视为一维张量。结果tensor的形状与索引相同。

参数介绍：

input：输入tensor。
indices：索引
接下来看用法：

src = torch.tensor([[4, 3, 5],
                    [6, 7, 8]])
torch.take(src, torch.tensor([0, 2, 5]))
tensor([ 4,  5,  8])
1
2
3
4

首先将input（这里为src）中的元素按照一维展开，然后在里面将index中的元素取出来形成一个新的tensor。index可以不只是一维tensor，可以是多维tensor：

注意： 返回的tensor不是视图，和原tensor不共享内存：更多计算机视觉与图形学相关资料，请关注微信公众号：计算机视觉与图形学实战

论文下载请在公众号内回复：paper如果您认为上面的内容对您有一定的价值，可以对我们进行小小的赞助，来支持我们的工作，因为后续打算构建自己的网站，谢谢：