Skip to content

💡 [REQUEST] - New recipe tutorial on calculating layer output dimensions #2926

@loganthomas

Description

@loganthomas

🚀 Describe the improvement or the new tutorial

This tutorial will help users understand how to transition from convolutional and pooling layers to linear layers in their models.

Learning objectives:

  • How to manually calculate the output dimensions after applying a convolution or pooling layer
  • How to print the shape of internal tensors for inspecting dimensionality changes in a model
  • How to use the torchinfo package to show output dimensions for all layers in a model

Existing tutorials on this topic

No response

Additional context

I created this draft (#2923) as a part of the PyTorch Docathon H1 2024 effort. I did not realize new tutorials weren't being accepted as part of the sprint and was asked to fill out an issue and convert the PR to a draft.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions