Skip to content

Add nn.Flatten Layer #2118

@VoVAllen

Description

@VoVAllen

This layer enables user to write conv layer and fc layer in one nn.Sequential model.

My implementation is at here:
https://gist.github.com/VoVAllen/5531c78a2d3f1ff3df772038bca37a83

Quite simple, but convenient for computer vision beginners.

Metadata

Metadata

Assignees

Labels

enhancementNot as big of a feature, but technically not a bug. Should be easy to fixmodule: 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