Squeeze2dLayer

class torchrl.modules.Squeeze2dLayer

用於卷積神經網路的擠壓層。

擠壓輸入張量的最後兩個單例維度。