报错：RuntimeError: Input, output and indices must be on the current device

运行程序突然碰到RuntimeError: Input, output and indices must be on the current device错误！！
环境：
pytorch1.7.1
python3.8

1. 问题描述

输入向量x在GPU上，index变量在CPU上导致

2. 解决办法

将index弄到GPU上，使其与x保持一致，代码如下：

index = index.to(torch.device("cuda"))

亲测有效O(∩_∩)O！！！