很容易在@pytorch 2.0的发布说明中错过，他们添加了一个小但有用的功能：torch.device，此前它只返回一个设备对象，现在可以用作上下文管理器。