SqueezeTransform
- class torchrl.envs.transforms.SqueezeTransform(*args, **kwargs)[source]
Removes a dimension of size one at the specified position.
- Parameters:
dim (int) – dimension to squeeze.
Removes a dimension of size one at the specified position.
dim (int) – dimension to squeeze.