Save a multiplication for the momentum update in the NN example