VIPRewardTransform

class torchrl.envs.transforms.VIPRewardTransform(*args, **kwargs)

A VIP transform to compute rewards based on embedded similarity.

This class will update the reward computation

forward(tensordict)

Reads the input tensordict, and for the selected keys, applies the transform.

reset(tensordict: TensorDictBase) → TensorDictBase

Resets a tranform if it is stateful.