pytorch:torch_scatter.scatter_max()
torch_scatter.scatter_max(src, index, dim=-1, out=None, dim_size=None, fill_value=None)根据index将src分组,求每一组中的最大值输出到outdim是维度from torch_scatter import scatter_maxsrc = torch.Tensor([[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]])index = torch.tensor([[4, 5, 4, 2
原创
2021-04-26 20:30:33 ·
2919 阅读 ·
0 评论