Everything is in the title.
I have a RaggedTensor of shape (nsamples, None, M1), and I would like to transpose the last two axes to get a RaggedTensor of shape (nsamples, M1, None).
tf.transpose does not work on ragged tensors, nor tf.linalg.matrix_transpose (as they are the same function).
Is there a way to do it purely in Tensorflow or do I have to do a (slow) for loop manually ?