reference/generated/tensordict.TensorDictBase.is_cuda

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

TensorDictBase.is_cuda

property TensorDictBase.is_cuda

Whether the tensordict has a CUDA device (False if it has no device).

Every tensordict class and every tensorclass has this method. For the other methods, see Methods by task.