C5W4 UNQ_4 EncoderLayer Wrong values when training=True

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