torch中以tensor作为索引取值

有:

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则是网络关于对应类的分类正确概率。