set_recurrent_mode

class torchrl.modules.set_recurrent_mode(mode: bool | Literal['recurrent', 'sequential'] | None = True)

用于设置 RNN 循环模式的上下文管理器。

参数:

mode (bool, "recurrent" 或 "sequential") – 在上下文管理器中使用的循环模式。"recurrent" 对应 mode=True，而 “sequential” 对应 mode=False。以循环模式“开启”执行的 RNN 假设数据是以时间批次的形式传入，否则假设 tensordict 中的每个数据元素是相互独立的。此上下文管理器的默认值为 True。默认循环模式是 None，即使用 RNN 的默认循环模式（参见 LSTMModule 和 GRUModule 构造函数）。

另请参阅

recurrent_mode`.

注意

默认情况下，所有 TorchRL 方法都使用 set_recurrent_mode(True) 进行装饰。