torch.Tensor.is_cuda

Tensor.is_cuda

Is True if the Tensor is stored on the GPU, False otherwise.