reference/generated/tensordict.TensorDictBase.auto_device_

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

TensorDictBase.auto_device

TensorDictBase.auto_device_() → Self

Automatically sets the device, if it is unique.

Returns: self with the edited device attribute.

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