You are free to write gate exam in any branch. But you should also check admission criteria. Each institute have different admission criteria you should go through brochure of all your dream college and check for branch in which you want to do mtech.

Feel free to ask me in the comment box if you are having any further doubts.

hope you find it helpful

Good Luck!