今天小编就为大家分享一篇浅谈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

在学习 CS231n中的NetworkVisualization-PyTorch任务，讲解了使用torch.gather函数，gather函数是用来根据你输入的位置索引 index，来对张量位置的数据进行合并，然后再输出。 其中 gather有两种使用方式，一种为 torch.gather 另一种为 对象.gather。 首先介绍 对象.gather import torch torch.manual_seed(2) #为CPU设置种子用于生成随机数，以使得结果是确定的 def gather