reference/generated/tensordict.TensorDictBase.dim

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

TensorDictBase.dim

TensorDictBase.dim() → int

Returns the number of batch dimensions, the same as batch_dims.

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