reference/generated/tensordict.TensorDictBase.__iter__

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

TensorDictBase.__iter__

TensorDictBase.__iter__() → Iterator[TensorDictBase]

Iterates over the first batch dimension of the tensordict.

Each item is a tensordict with the remaining batch dimensions, as returned by td.unbind(0). To iterate over the keys, use keys().

Raises:

TypeError – if the tensordict has no batch dimensions, as iter() does for a 0-d tensor.

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