SqueezeLayer

class torchrl.modules.SqueezeLayer(dims: Sequence[int] = (- 1,))

壓縮層。

壓縮輸入張量的給定單例維度。

引數：

dims (iterable) – 要壓縮的維度，預設為 (-1,)

forward(input: Tensor) → Tensor

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

應由所有子類覆蓋。

注意

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