title: ValueError: Unbatching a tensor is only supported for rank >= 1
body: <p>I am working on a sample project of California housing price problem and getting above error while training my model.</p>
  
  <p>Following this article 
  <a href="https://colab.research.google.com/notebooks/mlcc/first_steps_with_tensor_flow.ipynb?utm_source=mlcc&amp;utm_campaign=colab-external&amp;utm_medium=referral&amp;utm_content=firststeps-colab&amp;hl=en#scrollTo=pDIxp6vcU809" rel="nofollow noreferrer">https://colab.research.google.com/notebooks/mlcc/first_steps_with_tensor_flow.ipynb?utm_source=mlcc&amp;utm_campaign=colab-external&amp;utm_medium=referral&amp;utm_content=firststeps-colab&amp;hl=en#scrollTo=pDIxp6vcU809</a></p>