Squeeze2dLayer

class torchrl.modules.Squeeze2dLayer

卷積神經網路的壓縮層。

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