reference/generated/torchrl.envs.transforms.SqueezeTransform

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

SqueezeTransform

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

Removes a dimension of size one at the specified position.

Parameters:

dim (int) – dimension to squeeze.