pytorch Expected all tensors to be on the same device, but found at least two devices, cpu

在这里插入图片描述
出现上述,是说torch.mm函数中出现问题,输入x需要转到gpu上,对将x.mm(self.w)变成torch.mm(x.to(device), self.w)即可。

你可能感兴趣的:(python,pytorch,人工智能,python)