Rate this Page
★ ★ ★ ★ ★

TensorDictBase.float32#

TensorDictBase.float32() → Self#

Casts all tensors to torch.float32.

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