# Scale the modified vector a4
a4 /= keep_prob