variation on richard’s approach: simply return the xns you want, like:
return x4, x3, x2, x1

There is no rule/law that says that network modules need to return one and only one tensor.
(You can also return a list or a dictionary too by the way; I mean, you can return anything you like in fact; this is true for input parameters too by the way: you can have as many or as few input parameters as you want, and of any type you find works well for you)