It's better if you take Computer Science engineering. Because artificial intelligence needs a well enough knolwedge of programming and coding as well. You learn lots of languages and also computer organisation and architecture in your btech CSE syllabus.

So ya you must choose CSE. And for masters you can choose Aritificial intelligence stream.

Refer to the websites below for more detailed information..

https://www.careers360.com/courses/computer-science-engineering-course

https://engineering.careers360.com/tags/artificial-intelligence

Best wishes. Thank you.