from torch_scatter import scatter import torch src = (torch.rand(2, 6, 2)*4).int() index = torch.tensor([0, 1, 0, 1, 2, 1])
確定! 回上一頁