fully connected layer python code