Rate this Page
★ ★ ★ ★ ★

torch.Tensor.is_cuda#

Tensor.is_cuda#

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

On this page
Show Source
PyTorch Community
Language Bindings
PyTorch Libraries