TensorDictModuleWrapper

class tensordict.nn.TensorDictModuleWrapper(*args, **kwargs)

TensorDictModule 对象的包装类。

创建后，TensorDictModuleWrapper 的行为将与其包含的 TensorDictModule 完全一致，但被重写的方法除外。

参数：

td_module (TensorDictModule) – 待包装的操作符。

forward(*args: Any, **kwargs: Any) → TensorDictBase

定义每次调用时执行的计算。

应由所有子类重写。

注意

尽管前向传播的逻辑需要在本函数中定义，但之后应该调用 Module 实例而不是本函数本身，因为前者会负责运行注册的钩子（hooks），而后者会静默地忽略它们。