Can you explain a bit more about your answer ? isn’t it just calling the forward pass ? And how about this ?