How to use nn.Linear

######## i have fixed