reference/generated/tensordict.TensorDictBase.bfloat16

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

TensorDictBase.bfloat16

TensorDictBase.bfloat16() → Self

Casts all tensors to torch.bfloat16.

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