torch.device

A torch.device object represents the device on which an instance of a torch.Tensor is allocated.