Piège
Tentation : return A2 (la sortie 'utile'). Vidéo 9 dit explicitement 'on aura également besoin de A1 pour calculer les gradients'. Donc on retourne un dict {'A1': A1, 'A2': A2}. Sans A1, le backprop est cassé (besoin de $\sigma'(Z^{[1]}) = A^{[1]}(1-A^{[1]})$).