pytorch fully connected network example