Answers for "torch get device of tensor"

2

torch device

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
Posted by: Guest on December-01-2020

Python Answers by Framework

Browse Popular Code Answers by Language