SqueezeLayer

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

壓縮層。

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

參數:

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

forward(input: Tensor) → Tensor

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

應該被所有子類別覆寫。

注意

雖然正向傳遞的配方需要在這個函式中定義，但應該在此之後呼叫 Module 實例而不是這個函式，因為前者負責執行已註冊的 hooks，而後者會靜默地忽略它們。