DLS Course 2: Week 3: Exercise 6: Error with compute_total_loss function

You have to transpose labels and logits. Please check this.

PS: Sharing your code is not allowed. Next time, only share your full error.