C4_W2_Assig1_Ex2

It’s not a shape error.
The assert message says your code doesn’t give the correct values when “training” is set to False.