reference/ttd

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

TypedTensorDict

Warning

TypedTensorDict is experimental: its API can change in a minor release without a deprecation cycle.

This page lists TypedTensorDict, a TensorDictBase subclass whose fields you declare as class annotations, as in a TypedDict. A subclass adds fields to the fields of its parent, and from_tensordict() wraps any tensordict (H5, Redis, lazy stacks) without a copy. For a guide with examples, see TypedTensorDict.

TypedTensorDict has the methods of TensorDictBase, listed by task in Methods by task.

TypedTensorDict(*args, **kwargs)

A TensorDictBase subclass whose fields are declared as class annotations.

See also

• TypedTensorDict: declare and use a TypedTensorDict.

• Which container should I use?: tensorclass, TypedTensorDict, or TensorDict.

• Containers on storage backends: which containers work on which storage backends.