Fix dropout backward in TEST phase

This PR just add the option to pass gradients backwards during the `TEST` phase.
I just copied the code from the TRAIN mode, so don't know who add that in the first place. Maybe we can remove it.