让t
为形状为(m, n)
的Tensorflow张量。假设F
是p
个函数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。模式:渴望