在张量函数中应用沿一个轴的排列

Question

如何沿张量的单轴排列“维数”？你知道吗

类似于tf.transpose的东西，但在沿轴的“维度”层次上，而不是在轴的层次上。你知道吗

要随机排列它们（沿着第一个轴），就有tf.random.shuffle，要移动它们，就有tf.roll。但是我找不到一个更一般的函数可以应用任何给定的排列。你知道吗

Answer 1

tf.gather可用于此目的。事实上，它更一般，因为它作为输入之一的索引不需要表示置换。你知道吗