InitTracker

class torchrl.envs.transforms.InitTracker(init_key: str = 'is_init')

重置追蹤器。

此轉換使用重置追蹤器條目填充步進/重置 tensordict，該條目在呼叫 reset() 時被設定為 True。

引數:

init_key (NestedKey, 可選) – 用於追蹤器條目的鍵。在存在多個 _reset 標誌的情況下，此鍵用作每個標誌的葉子替換。

示例

>>> from torchrl.envs.libs.gym import GymEnv
>>> env = TransformedEnv(GymEnv("Pendulum-v1"), InitTracker())
>>> td = env.reset()
>>> print(td["is_init"])
tensor(True)
>>> td = env.rand_step(td)
>>> print(td["next", "is_init"])
tensor(False)

forward(tensordict: TensorDictBase) → TensorDictBase

讀取輸入的 tensordict，並對選定的鍵應用轉換。

transform_observation_spec(observation_spec: TensorSpec) → TensorSpec¶

轉換 observation spec，使結果 spec 與轉換對映匹配。

引數:

observation_spec (TensorSpec) – 轉換前的 spec

返回:

轉換後預期的 spec