tf.keras.ops.moveaxis

Move axes of a tensor to new positions.

Other axes remain in their original order.

x Tensor whose axes should be reordered.
source Original positions of the axes to move. These must be unique.
destination Destinations positions for each of the original axes. These must also be unique.

Tensor with moved axes.