Why is mdl.forward(batch_xs) a bug? What incorrect behaviour could it result in?
mdl.forward(batch_xs)