在TensorFlow中沿一个轴应用置换

Question

如何在张量的单轴上排列“尺寸”？

类似于tf.transpose的东西，但沿轴的“维度”级别，而不是轴的级别。

要随机排列它们（沿第一轴），tf.random.shuffle，然后移动tf.roll。但是我找不到更通用的函数来应用任何给定的排列。

Answer 1

tf.gather可以用于此目的。实际上，它甚至更通用，因为它作为输入之一使用的索引不需要表示排列。