TensorDictModuleWrapper

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

用於包裝 TensorDictModule 物件的類。

建立後，TensorDictModuleWrapper 的行為將與其包含的 TensorDictModule 完全相同，但被覆蓋的方法除外。

引數:

td_module (TensorDictModule) – 要包裝的運算子。

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

定義每次呼叫時執行的計算。

所有子類都應重寫此方法。

注意

儘管前向傳播的實現需要在此函式中定義，但您應該在之後呼叫 Module 例項而不是此函式，因為前者會處理註冊的鉤子，而後者則會靜默忽略它們。