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

时间:2019-06-04 12:22:01

标签: python tensorflow parallel-processing

t为形状为(m, n)的Tensorflow张量。假设Fp个函数f_1, f_2, ...的列表,其中每个函数取一个张量并输出相同形状的张量。

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

请注意,这是map函数的逆设置,其中我们将1个函数应用于元素集合。在这里,我们需要将功能集合应用于1个元素。此外,我们需要并行进行此操作。

天真的方法(非并行)是将通过for循环(或列表理解)生成的这些计算的结果进行堆叠。

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

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

1 个答案:

答案 0 :(得分:0)

假设r1.13并不急于执行tf.parallel_stack应该可以解决问题。