example - Swap tensor de ejes en tensorflow