SafeProbabilisticTensorDictSequential

class torchrl.modules.tensordict_module.SafeProbabilisticTensorDictSequential(*args, **kwargs)

是 tensordict.nn.ProbabilisticTensorDictSequential 的子类，接受 `TensorSpec` 参数来控制输出域。

与 TensorDictSequential 类似，但强制要求序列中的最后一个模块是 ProbabilisticTensorDictModule，并暴露 `get_dist` 方法以便从 ProbabilisticTensorDictModule 中恢复分布对象。

参数：:

• modules (TensorDictModules 可迭代对象) – TensorDictModule 实例的有序序列，以 ProbabilisticTensorDictModule 结尾，按顺序运行。

• partial_tolerant (bool, 可选) – 如果为 `True`，输入 tensordict 可以缺少一些输入键。在这种情况下，只有那些在现有键下可以执行的模块会被执行。此外，如果输入 tensordict 是 tensordict 的惰性堆栈 *并且* partial_tolerant 为 `True` *并且* 堆栈没有所需的键，那么 TensorDictSequential 将扫描子 tensordict，查找包含所需键的那些（如果存在）。