SqueezeTransform

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

Removes a dimension of size one at the specified position.

Parameters:

squeeze_dim (int) – dimension to squeeze.