C5W1A3 music_inference_model problem

Guys I’ve been investigating but I can address the problem. I think is something with the shapes. You can see it on the image:

Thanks in advance!

Search the forum for the phrase “multiple inbound nodes”. There are several threads on this.

I’ve just spotted. I think the argmax part is very confusing. I needed to specify the axis required.

Thanks!