如何将多个张量流函数并行应用于同一个张量？

Question

让t 成为形状为(m, n) 的Tensorflow 张量。让F 成为p 函数f_1, f_2, ... 的列表，其中每个函数都采用一个张量并输出一个相同形状的张量。

我需要计算形状为(p, m, n) 的张量T 使得T[i]=f_i(t)。

请注意，这是 map 函数的逆设置，我们将 1 个函数应用于元素集合。在这里，我们需要将一组函数应用于 1 个元素。此外，我们需要并行这样做。

朴素的方法（非并行）是将通过 for 循环（或列表推导）生成的这些计算的结果堆叠起来。

T = tf.stack([
  f(t) for f in F
])

Tensorflow 版本：2.0.0a。模式：渴望

Answer 1

假设 r1.13 没有急切执行 tf.parallel_stack 应该可以解决问题。