torch.Tensor.is_cuda

Tensor.is_cuda

如果张量存储在GPU上，则为True，否则为False。