总说 一个非常有用的函数,主要是用于“group index”的操作。 先安装一下 https://github.com/rusty1s/pytorch_scatter from torch_scatter import scatter import torch src = (torch.rand(2, 6, 2)*4).int()