But only during the forward pass of the neural network?