pytorch create fully connected layer