Torch tensor.gather()學習

最近在學習Pytorch的時候遇見了一個很棘手的函數gather，之前也轉載了一個博客，但是也似懂非懂的，今天在此總結下，以便後續需要~ 首先貼一下官方的介紹： torch.gather(input, dim, index, out=None) → Tensor Gathers values along an axis specified by dim.（沿着dim指定的軸聚集 s值。） For

>>阅读原文<<