VIPRewardTransform

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

一个基于嵌入相似性计算奖励的 VIP Transform。

此类将更新奖励计算

forward(tensordict)

读取输入的 tensordict，并对选定的键应用 Transform。

transform_input_spec(input_spec: TensorSpec) → TensorSpec

转换输入 Spec，使结果 Spec 与 Transform 映射匹配。

参数:

input_spec (TensorSpec) – 应用 Transform 之前的 Spec

返回:

应用 Transform 之后预期的 Spec