Compute_Loss

Course Q&A Deep Learning Specialization Improving Deep Neural Networks: Hyperparameter tun
8

You have to transpose logits and labels. Check this.