torch.Tensor.is_cuda

Tensor.is_cuda

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