As a Sensei, I have people training with me who have reached higher grades and then stopped.
In one case the student trained with me a few times and, after speaking with my Sensei, I told him we were happy for him to wear the colored belt he had previously achieved. Other students just wear their belt without issue. It does depend a bit on how long the break has been, but generally, as long as you trained in the same style, you should be able to wear your belt. This enables you to stand with other students of that belt level and pick things up that lower belt students don't know yet but that you would have done before.
In any case, safest way is to talk to your Sensei. Don't be surprised if you pick things up quickly. Welcome back.
Barbah