torch.gather(input, dim, index, out=None) 和torch.scatter_(dim, index, src)是一对作用相反的方法. 先来看torch.gather, 核心操作其实就是这样:.
確定! 回上一頁