今天小编就为大家分享一篇浅谈Pytorch中的torch.gather函数的含义，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

pytorch中的gather函数 pytorch比tensorflow更加编程友好，所以准备用pytorch试着做最近要做的一些实验。 立个flag开始学习pytorch，新开一个分类整理学习pytorch中的一些踩到的泥坑。 今天刚开始接触，读了一下documentation，写一个一开始每太搞懂的函数gather b = torch.Tensor([[1,2,3],[4,5,6]]) print b index_1 = torch.LongTensor([[0,1],[2,0]]) i