Skip to content

BatchNorm{1-2-3}d are redundant #2628

@fmassa

Description

@fmassa

Since #2020, nn.BatchNorm1d/nn.BatchNorm2d/nn.BatchNorm3d are completely redundant, as there are no more size checks that are performed in the input tensor
I see two options here:

  • add the size-checks back for each class;
  • deprecate BatchNorm{1-2-3}d and use instead BatchNorm.

We already have only a single functional interface for BatchNorm, so option 2 might be worth considering.

cc. @alykhantejani who removed the size checks

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: nnRelated to torch.nntriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions