reference/generated/tensordict.utils.DeviceType

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

DeviceType

tensordict.utils.DeviceType

The type of the device arguments.

A torch.device, a str such as "cpu" or "cuda:0", or the index of a CUDA device.

alias of Union[device, str, int]