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。以 recurrent_mode “on” 执行的 RNN 假定数据以时间批次的形式输入，否则假定 tensordict 中的每个数据元素都是独立的。此上下文管理器的默认值为 True。默认的递归模式是 None，即使用 RNN 的默认递归模式（请参阅 LSTMModule 和 GRUModule 构造函数）。

另请参阅

recurrent_mode`.

注意

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