主要介绍了Pytorch中index_select() 函数的实现理解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

函数形式： index_select( dim, index ) 参数： dim：表示从第几维挑选数据，类型为int值； index：表示从第一个参数维度中的哪个位置挑选数据，类型为torch.Tensor类的实例； 刚开始学习pytorch，遇到了index_select(),一开始不太明白几个参数的意思，后来查了一下资料，算是明白了一点。 a = torch.linspace(1, 12, steps=12).view(3, 4) print(a) b = torch.i