It works ! thank you very much ! but I still have some question：
1.what do the 131072 and 8 mean?
2.Is it means that I need to use view function to flatten activation before passing to Linear layer every time ?