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