I truly feel an excellent illustration (common situation early on in pytorch before the flatten layer was official included was this typical code): This way you could have One more button in precisely the same kind that does post the shape, in case that's essential. I also Believe this is https://emilianohszfk.blogdon.net/a-simple-key-for-learn-more-unveiled-55508729