pytorch索引查找 index_select

最新推荐文章于 2025-07-21 00:28:47 发布

AI算法网奇

最新推荐文章于 2025-07-21 00:28:47 发布

阅读量3.2w

点赞数 7

CC 4.0 BY-SA版权

分类专栏： pytorch知识宝典

本文链接：https://blog.youkuaiyun.com/jacke121/article/details/83044660

pytorch知识宝典专栏收录该内容

512 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

这篇博客介绍了PyTorch中的index_select函数，通过示例展示了如何在维度1上根据给定的索引选取数据。文章解释了参数含义，并给出了不同索引选择结果的示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

index_select

anchor_w = self.FloatTensor(self.scaled_anchors).index_select(1, self.LongTensor([0]))

参数说明：index_select(x, 1, indices)

1代表维度1，即列，indices是筛选的索引序号。

例子：

import torch


x = torch.linspace(1, 12, steps=12).view(3,4)

print(x)
indices = torch.LongTensor([0, 2])
y = torch.index_select(x, 0, indices)
print(y)

z = torch.index_select(x, 1, indices)
print(z)

z = torch.index_select(y, 1, indices)
print(z)

结果：

tensor([[ 1., 2., 3., 4.],
[ 5., 6., 7., 8.],
[ 9., 10., 11., 12.]])
tensor([[ 1., 2., 3., 4.],
[ 9., 10., 11., 12.]])
tensor([[ 1., 3.],
[ 5., 7.],
[ 9., 11.]])
tensor([[ 1., 3.],
[ 9., 11.]])