Week 3 - Exercise 6 - Compute Total Loss

Hi @Syed_Hamza_Tehseen

Here is one of the points listed in the implementation instruction:

If you read the comments on the input arguments:

logits -- output of forward propagation (output of the last LINEAR unit), of shape (6, num_examples)
labels -- "true" labels vector, same shape as Z3

It is because the logits and labels are not in the shape that is expected by the categorical_crossentropy, so, in order for it to work correctly, we need to transpose logits and labels.