Pytorch之gather的用法(易懂的方式解释)
百度了一圈gather的用法,看的一知半解,结合了几篇博客的讲解,终于理解了这个的用法,记录下来,用于以后忘记的时候自己可以快速复习,同时不懂得小伙伴也可以参考下我这得理解,或许能帮助到你!!!首先了解下函数中的参数:torch.gather(input, dim, index, out=None) → TensorParameters: input (Tensor) – The source tensor dim (int) – The axis along wh
原创
2020-06-20 16:53:58 ·
4763 阅读 ·
4 评论