Rate this Page
★ ★ ★ ★ ★

TensorDictBase.depth#

property TensorDictBase.depth: int#

the maximum number of levels of sub-tensordicts.

A tensordict with no sub-tensordict has depth 0.

Examples

>>> from tensordict import TensorDict
>>> TensorDict({"a": 0}).depth
0
>>> TensorDict({"a": {"b": {"c": 0}}}).depth
2
Type:

Returns the depth of the tensordict

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