Rate this Page
★ ★ ★ ★ ★

TensorDictBase.dtype#

property TensorDictBase.dtype#

Returns the dtype of the leaves, or None if the leaves have different dtypes.

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