Squeeze2dLayer

class torchrl.modules.Squeeze2dLayer

用于卷积神经网络的压缩层。

压缩输入张量的最后两个单一维度。