torch.Tensor.is_cuda

Tensor.is_cuda

如果 Tensor 儲存在 GPU 上，則為 True，否則為 False。