SqueezeTransform

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

在指定位置移除大小為一的維度。

引數:

dim (int) – 要擠壓的維度。