python - 在 TensorFlow 中沿一个轴应用排列

Question

如何沿张量的单个轴排列“维度”？

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

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

标签： pythontensorflow

Answer 1

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