Make sure the Python interpreter selected in VS Code matches the one with which you installed tensorflow.

By default you will find the selected Python interpreter in the bottom right of VS Code if you have a Python file selected. More info: https://code.visualstudio.com/docs/python/environments.