reference/generated/tensordict.TensorDictBase.ndimension

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

TensorDictBase.ndimension

TensorDictBase.ndimension() → 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.