来源：https://stackoverflow.com/questions/57933781/how-pytorch-tensor-get-the-index-of-elements 标签 python pytorch torch tensor
