TensorDictBase.cpu#
- TensorDictBase.cpu(**kwargs) Self#
Casts a tensordict to CPU.
This function also supports all the keyword arguments of
to().
Every tensordict class and every tensorclass has this method. For the other methods, see Methods by task.