Pytorch Count Number Of Parameters In Logistic Regression