InitTracker

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

重置跟踪器。

此转换使用在调用 reset() 时设置为 True 的重置跟踪器条目填充 step/reset tensordict。

参数:

init_key (NestedKey, optional) – 用于跟踪器条目的键。在多个 _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 (TensorSpec) – 转换前的规范

返回:

转换后预期的规范