y
Out[34]: tensor([0, 2])
y_hat
Out[35]:
tensor([[0.1000, 0.3000, 0.6000],
[0.3000, 0.2000, 0.5000]])
y_hat.gather(1, y.view(-1, 1))
聚合方向y_hat的维度1,聚合位置:
a[0][y.view(-1,1)[0]] = 0.1
a[1][y.view(-1,1)[1]] = 0.5
y
Out[34]: tensor([0, 2])
y_hat
Out[35]:
tensor([[0.1000, 0.3000, 0.6000],
[0.3000, 0.2000, 0.5000]])
y_hat.gather(1, y.view(-1, 1))
聚合方向y_hat的维度1,聚合位置:
a[0][y.view(-1,1)[0]] = 0.1
a[1][y.view(-1,1)[1]] = 0.5