Introduction to tensor flow

Solved for @Kiranmayee.

Never hard-code dimensions unless the exercise explicitly tells you to do so.

Good luck with the rest of the specialization :slight_smile: