I\'m trying to make categorical cross entropy loss function to better understand intuition behind it. So far my implementation looks like this: