index_select

可以快速上手的开发文档

#index_select

pytorch索引查找 index_select的例子

index_selectanchor_w=self.FloatTensor(self.scaled_anchors).index_select(1,self.LongTensor([0]))参数说明:index_sele..

Pytorch中index_select() 函数的实现理解

函数形式:index_select(dim,index)参数:dim:表示从第几维挑选数据,类型为int值;index:表示..