今天小编就为大家分享一篇pytorch索引查找 index_select的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

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.L