Hello there , first of all congrats for your achivement. I would like to suggest you should opt for Mtech in ee only as there are many pre knowledge is required which are tought in Btech cse curriculam.

But if you are intersted in AI and machine learning tou can opt for online training such as codeblocks, codingNinja offers, also recently Bits Pilani is also offering short term course in AI&Machine learning.