With our templates z (i) in hand, we can then restrict the model to using (one of) them during generation.
given a new input x, we may generate by computinĝ which gives us a generationŷ (i) for each extracted template z (i) .