in the dropout python code

[Dungyichao]

Post: Drop Out Regularization
original code
Z1 = np.dot(W1.T, X) + b1 A = np.tanh(Z1) D = np.random.rand(A.shape[0], A.shape[1]) D = D < keep_prob A * D A = A / keep_prob Z2 = np.dot(W2.T, X) + b2

in the last line, I think it should be
Z2 = np.dot(W2.T, A) + b2