如何将torch.device(“torch.cuda.is_available”如果torch.cuda.is_available()“torch.cuda.is_available”)写成完整的if write语句？

Question

我是个初学者，我想把这个语句作为一个整体输入，如果是其他语句：-

torch.device('cuda' if torch.cuda.is_available() else 'cpu')

有人能帮我吗？

Answer 1

下面是一个完整的if-else语句代码：

torch.device('cuda' if torch.cuda.is_available() else 'cpu')

if torch.cuda.is_available():
    torch.device('cuda')
else:
    torch.device('cpu')

由于您可能希望稍后存储该设备，因此您可能需要这样的内容：

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

if torch.cuda.is_available():
    device = torch.device('cuda')
else:
    device = torch.device('cpu')

下面是关于Python：https://stackoverflow.com/a/2802748/13985765中三元操作符的文章和讨论

从该职位：

value_when_true if condition else value_when_false