InitTracker

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

重置跟踪器。

此转换使用重置跟踪器条目填充步骤/重置张量字典，每当调用 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

读取输入张量字典，并对选定的键应用转换。

transform_observation_spec(observation_spec: TensorSpec) → TensorSpec

转换观察规范，以便生成的规范与转换映射匹配。

参数：

observation_spec (TensorSpec) – 转换前的规范

返回值：

转换后的预期规范