SqueezeTransform

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

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

參數:

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