C5W4 UNQ_4 EncoderLayer Wrong values when training=True

I’ve looked through the search archives for this issue and repeatedly rechecked the instructions, but still can’t seem to find out where the problem is. What should I be looking for?

Adding print statement might help to find the bug. Please check this.