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是筛选的索引序号。
例子：
importtorch 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.]])
以上这篇pytorch索引查找index_select的例子就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
热门推荐
10 小红书平安祝福语简短
11 生日祝福语大全女孩简短
12 收生日红包祝福语 简短
13 领证幽默祝福语简短
14 法考面试祝福语简短
15 老哥出门祝福语简短语
16 送灯祝福语简短独特
17 幼儿狗年祝福语大全简短
18 好听的元旦简短祝福语