Thank You Sir,
But I am building a model  Logistic regression with “m” parameters using pytorch “nn” module…
if we use output_dim=2, we are getting 2*m weights which is not required.
If I changed output_dim=1, I am getting an error…
Is there any other way to develop logistic regression model in pytorch  without nn module?