TensorDictModuleWrapper

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

TensorDictModule 物件的包裝類。

建立後，TensorDictModuleWrapper 的行為將與其包含的 TensorDictModule 完全一致，但被重寫的方法除外。

引數：

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

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

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

應由所有子類重寫。

注意

儘管前向傳播的邏輯需要在本函式中定義，但之後應該呼叫 Module 例項而不是本函式本身，因為前者會負責執行註冊的鉤子（hooks），而後者會靜默地忽略它們。