SignTransform

class torchrl.envs.transforms.SignTransform(in_keys=None, out_keys=None, in_keys_inv=None, out_keys_inv=None)

計算 TensorDict 值符號的 transform。

此 transform 讀取 in_keys 和 in_keys_inv 中的張量，計算其元素的符號，並將結果符號張量分別寫入 out_keys 和 out_keys_inv 中。

引數：

• in_keys (list of NestedKeys) – 輸入條目（讀取）

• out_keys (list of NestedKeys) – 輸入條目（寫入）

• in_keys_inv (list of NestedKeys) – 在 inv() 呼叫期間讀取的輸入條目。

• out_keys_inv (list of NestedKeys) – 在 inv() 呼叫期間寫入的輸入條目。

示例

>>> from torchrl.envs import GymEnv, TransformedEnv, SignTransform
>>> base_env = GymEnv("Pendulum-v1")
>>> env = TransformedEnv(base_env, SignTransform(in_keys=['observation']))
>>> r = env.rollout(100)
>>> obs = r["observation"]
>>> assert (torch.logical_or(torch.logical_or(obs == -1, obs == 1), obs == 0.0)).all()

transform_observation_spec(observation_spec: TensorSpec) → TensorSpec

轉換觀察規範，使結果規範與 transform 對映匹配。

引數：

observation_spec (TensorSpec) – transform 之前的規範

返回：

transform 後的預期規範

transform_reward_spec(reward_spec: TensorSpec) → TensorSpec

轉換獎勵規範，使結果規範與 transform 對映匹配。

引數：

reward_spec (TensorSpec) – transform 之前的規範

返回：

transform 後的預期規範