Solved by chinmay5 in post #2 
              
              
                The linear layer only does a linear transformation and does not include a non-linearity inbuilt. So, the second one is actually creating a MLP in true sense.