SqueezeTransform

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

移除指定位置的大小为 1 的维度。

参数:

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