Update 01_core.ipynb

[khushpatelmd]

Thank you for this amazing course. I found one change dim =1 not dim=0 for the softmax. I tried creating a smaller dummy example to understand it.