Squeeze2dLayer

class torchrl.modules.Squeeze2dLayer

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

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