gpu加速后不能使用one_hot

mnist数据集label进行to(device)处理后就不能用one_hot了，有人知道为什么？