SqueezeTransform

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

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

参数:

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