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