You will need to flatten this to provide it to the fully linked layer. Therefore you convey to PyTorch to reshape the tensor you received to get precise amount of columns and convey to it to determine the quantity of rows by alone. I like to recommend listening to audio https://gunnerzitua.estate-blog.com/38453065/the-smart-trick-of-view-company-profile-that-nobody-is-discussing