You can probably go back, but...
How will dropping out improve things? It sounds like a poor approach unless you have some positive options to make your situation better. How will dropping out improve your study skills, say. It won't improve your already finished undergraduate degree.
People do leave and return. That isn't a huge problem most places. But when you want to return, people will ask you about your suitability and the probability of your success.
Let me suggest you think about whether staying and working on study habits and work process isn't a better path for you than leaving.
Of course, you may have other reasons that you don't state here that make leaving for a while a better plan. But if you go, have a plan, not just a hope, for your return if that is your intention.