Table of Contents

Shortcuts

reference/generated/torchrl.modules.Squeeze2dLayer

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

Squeeze2dLayer¶

class torchrl.modules.Squeeze2dLayer[source]¶

Squeezing layer for convolutional neural networks.

Squeezes the last two singleton dimensions of an input tensor.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources