Beautify the conditional handler

So far, it had to restore indexing information on the visitor
for the second branch. This can be avoided by index pushdown,
just as we do with the sum of two tensors.