Why do we need the nn.Flatten() layer? Why not going directly to nn.Linear ? #774
Replies: 1 comment
-
I found the answer myself. For those who have the same question : |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
I am still learning and created some CNN models for different classification use cases. I love Daniel's way of coding compared to others but the one thing I don't understand is why do we need the flatten layer instead of directly going to nn.Linear() as others do?
Beta Was this translation helpful? Give feedback.
All reactions