Hi,
It is not very clear from your question what kind of objects input and output are. But adding an extra entry at the beginning will do this: [None,] + output for a list for example. Or torch.cat([torch.tensor(whatever_is_the_right_size), output], 0) for Tensors.