用于tensorflow的批处理操作包装器

Question

是否有任何方法可以“包装”张量流操作，因此它将重复batch_size次？

例如，假设我要实现一个批处理tf.roll操作。 我从一批矩阵（假设形状为（32，10，10））和移位指示器（我们要从批次中移出单个矩阵-形状为（32，1））开始。

真正实现它的最佳方法是什么，因为我用python for loop尝试了它，但是它没有用。