有:
y=torch.tensor([1,2,2])
y1=torch.tensor([
[0.1,0.2,0.3],
[0.4,0.5,0.6],
[0.7,0.8,0.9]
])
y1[[0,1,2],y] = tensor([0.2, 0.6, 0.9])
以上可将y1[[0,1,2],y]理解为:在y1中,以y的值作为下标位置索引取值,位置上:行=[0,1,2],
列=y ([1,2,2]) 以此位置做取值操作,故输出结果如上,在深度学习中,y常做分类编号,y1则是网络关于对应类的分类正确概率。