python分析集聚程度_Python中的聚集和分散函数,pytorch,gather,scatter

import torch

a = torch.randint(0, 30, (2, 3, 5))

print(a)

'''

tensor([[[ 18., 5., 7., 1., 1.],

[ 3., 26., 9., 7., 9.],

[ 10., 28., 22., 27., 0.]],

[[ 26., 10., 20., 29., 18.],

[ 5., 24., 26., 21., 3.],

[ 10., 29., 10., 0., 22.]]])

'''

index = torch.LongTensor([[[0,1,2,0,2],

[0,0,0,0,0],

[1,1,1,1,1]],

[[1,2,2,2,2],

[0,0,0,0,0],

[2,2,2,2,2]]])

b = torch.gather(a, 1,index)

print(b)

'''

tensor([[[ 18., 26., 22., 1., 0.],

[ 18., 5., 7., 1., 1.],

[ 3., 26., 9., 7., 9.]],

[[ 5., 29., 10., 0., 22.],

[ 26., 10., 20., 29., 18.],

[ 10., 29., 10., 0., 22.]]])

当dim=1时:

out[0][0][0] = out[0][index[0][0][0]][0] = out[0][0][0] = 18

out[0][0][1] = out[0]