SqueezeTransform

class torchrl.envs.transforms.SqueezeTransform(*args, **kwargs)

在指定位置删除大小为一的维度。

参数：

squeeze_dim (int) – 要压缩的维度。