torch.Tensor.is_cuda

Tensor.is_cuda

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