Table of Contents

Shortcuts

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