torch.get_default_device

torch.get_default_device() [source]

Gets the default torch.Tensor to be allocated on device

Return type

device