FiniteTensorDictCheck¶
- class torchrl.envs.transforms.FiniteTensorDictCheck[source]¶
This transform will check that all the items of the tensordict are finite, and raise an exception if they are not.
- forward(tensordict: TensorDictBase) TensorDictBase ¶
Reads the input tensordict, and for the selected keys, applies the transform.