FiniteTensorDictCheck

class torchrl.envs.transforms.FiniteTensorDictCheck

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.