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