You need to flatten this to give it for the fully connected layer. Which means you tell PyTorch to reshape the tensor you received to acquire specific variety of columns and tell it to decide the quantity of rows by itself. You may have picked textbooks which can be aligned https://reidcmsyc.blogginaway.com/40271077/considerations-to-know-about-explore-services