You have to flatten this to provide it for the fully linked layer. So you convey to PyTorch to reshape the tensor you attained to possess particular variety of columns and tell it to make your mind up the quantity of rows by itself. A great number of guides, so https://felixxmsib.newsbloger.com/40187951/the-smart-trick-of-learn-more-that-no-one-is-discussing